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 (ArrayLike) – a PRNG key (from key, split, fold_in).

  • data (IntegerArray) – a 32-bit integer representing data to be folded into the key.

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.

Return type:

Array