triton.language.rand

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

给定一个 seed 标量和一个 offset 块，返回一个由随机 float32 组成的块，这些随机数遵循 \(U(0, 1)\) 分布。

参数：

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

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