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 (fromkey,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 isstatistically safe for producing a stream of new pseudo-random values.

Return type:

Array