triton.language.rand

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

给定一个seed标量和一个offset块， 返回一个在\(U(0, 1)\)范围内的随机float32块。

Parameters:

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

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