generate random bits of desired shape and dtype
Usage
nv_rng_bit_generator(
initial_state,
rng_algorithm = "THREE_FRY",
dtype,
shape_out
)
Arguments
- initial_state
state seed
- rng_algorithm
one of 'DEFAULT', 'THREE_FRY', 'PHILOX'
- dtype
datatype of output
- shape_out
output shape