triton.language.randint4x

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

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

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

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

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