triton.language.rand

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

Given a seed scalar and an offset block, returns a block of random float32 in \(U(0, 1)\).

Parameters:

• seed – The seed for generating random numbers.

• offsets – The offsets to generate random numbers for.