Script.randint

Contents

Script.randint

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

Generate a block of random int32 using Philox-4x32 PRNG.

Given a seed scalar and an offset register tensor, returns a single register tensor of random uint32 values.

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 uint32 values with the same shape as offset.

Return type:

RegisterTensor

Notes

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