jax.random.bits#

jax.random.bits(key, shape=(), dtype=None)[source]#

Sample uniform bits in the form of unsigned integers.

Parameters:

  • key (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – a PRNG key used as the random key.

  • shape (Sequence[int]) – optional, a tuple of nonnegative integers representing the result shape. Default ().

  • dtype (Union[str, type[Any], dtype, SupportsDType, None]) – optional, an unsigned integer dtype for the returned values (default uint64 if jax_enable_x64 is true, otherwise uint32).

Return type:

Array

Returns:

A random array with the specified shape and dtype.