Primitive RNG Bit Generator — nvl_rng_bit_generator • anvil

Generates random bits using the specified algorithm.

Usage

nvl_rng_bit_generator(
  initial_state,
  rng_algorithm = "THREE_FRY",
  dtype,
  shape_out
)

Arguments

initial_state: (tensorish)
RNG state tensor.
rng_algorithm: (character(1))
Algorithm name (default "THREE_FRY").
dtype: (character(1) | stablehlo::TensorDataType)
Data type.
shape_out: (integer())
Output shape.

Value

List of new state and random tensor.