jax.random.PRNGKey#
- jax.random.PRNGKey(seed, *, impl=None)[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 optional
impl
argument or, otherwise, by thejax_default_prng_impl
config flag.- Parameters:
seed (int | ArrayLike) – a 64- or 32-bit integer used as the value of the key.
impl (PRNGSpecDesc | None) – optional string specifying the PRNG implementation (e.g.
'threefry2x32'
)
- Return type:
KeyArray
- Returns:
A PRNG key, consumable by random functions as well as
split
andfold_in
.