triton.language.randint4x

triton.language.randint4x(seed, offset, n_rounds: constexpr = 10)

给定一个seed标量和一个offset块，返回四个随机int32块。

这是访问Triton的Philox伪随机数生成器最高效的入口点。

Parameters:

  • seed – 用于生成随机数的种子。

  • offsets – 生成随机数的偏移量。