jax.random.fold_in#

jax.random.fold_in(key, data)[source]#

Folds in data to a PRNG key to form a new PRNG key.

Parameters
  • key (Union[Array, PRNGKeyArray]) – a PRNG key (from PRNGKey, split, fold_in).

  • data (int) – a 32bit integer representing data to be folded in to the key.

Return type

Union[Array, PRNGKeyArray]

Returns

A new PRNG key that is a deterministic function of the inputs and is statistically safe for producing a stream of new pseudo-random values.