Script.rand

Contents

Script.rand

Script.rand(seed, offset, n_rounds=10)

Generate a block of random float32 in U(0, 1) using Philox-4x32 PRNG.

Given a seed scalar and an offset register tensor, returns a register tensor of random float32 values uniformly distributed in [0, 1).

Parameters:
  • seed (Expr | int) – The seed for generating random numbers (uint64 scalar).

  • offset (RegisterTensor) – The offsets to generate random numbers for (uint32).

  • n_rounds (int) – Number of Philox rounds (default 10).

Returns:

ret – A register tensor of random float32 values in [0, 1) with the same shape as offset.

Return type:

RegisterTensor

Notes

  • Thread group: Can be executed by any sized thread group.