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 (Union[int, Array]) – a 64- or 32-bit integer used as the value of the key.

Return type

Union[Array, PRNGKeyArray]

Returns

A PRNG key, consumable by random functions as well as split and fold_in.