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[Any, 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[Any, 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.