jax.random.bits

jax.random.bits(key, shape=(), dtype=None, *, out_sharding=None)

Sample uniform bits in the form of unsigned integers.

Parameters:

• key (ArrayLike) – a PRNG key used as the random key.

• shape (Shape) – optional, a tuple of nonnegative integers representing the result shape. Default ().

• dtype (DTypeLikeUInt | None | None) – optional, an unsigned integer dtype for the returned values (default uint64 if jax_enable_x64 is true, otherwise uint32).

Returns:

A random array with the specified shape and dtype.

Return type:

Array