jax.random.key_data# jax.random.key_data(keys)[source]# Recover the bits of key data underlying a PRNG key array. Parameters: keys (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – Return type: Array