jax.random.PRNGKey
-
jax.random.PRNGKey(seed)[source]
Create a pseudo-random number generator (PRNG) key given an integer seed.
The resulting key carries the default PRNG implementation, as
determined by the jax_default_prng_impl config flag.
- Parameters
seed (int) – a 64- or 32-bit integer used as the value of the key.
- Return type
Union[Any, PRNGKeyArray]
- Returns
A PRNG key, consumable by random functions as well as split
and fold_in.
