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 (ndarray) – a PRNGKey (an array with shape (2,) and dtype uint32).

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

Return type

ndarray

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.