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 (defaultuint64
ifjax_enable_x64
is true, otherwiseuint32
).
- Return type:
- Returns:
A random array with the specified shape and dtype.