triton.language.rand

triton.language.rand(seed, offset)

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.