Ctrl+K

jax.random.bits

Contents

jax.random.bits#

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

Sample uniform bits in the form of unsigned integers.

Parameters:

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

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

  • dtype (DTypeLikeUInt | 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.

Contents