jax.random.fold_in

Contents

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:
Return type:

Array

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.