triton.language.randint4x

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

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

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

参数:

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

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