jax.random.key_data#

jax.random.key_data(keys)[source]#

Recover the bits of key data underlying a PRNG key array.

Parameters:

keys (ArrayLike)

Return type:

Array