triton.language.rand

triton.language.rand(seed, offset, n_rounds: triton.language.core.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.