jax.random.key#
- jax.random.key(seed, *, impl=None)[source]#
Create a pseudo-random number generator (PRNG) key given an integer seed.
The result is a scalar array with a key that indicates the default PRNG implementation, as determined by the optional
impl
argument or, otherwise, by thejax_default_prng_impl
config flag.- Parameters:
- Return type:
- Returns:
A scalar PRNG key array, consumable by random functions as well as
split
andfold_in
.