Warning

This page was created from a pull request.

jax.random.fold_in

jax.random.fold_in(key, data)

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

Parameters

• key – a PRNGKey (an array with shape (2,) and dtype uint32).

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

Returns

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