jax.extend.random.define_prng_impl

jax.extend.random.define_prng_impl(*, key_shape, seed, split, random_bits, fold_in, name='<unnamed>', tag='?')

Parameters:

• key_shape (tuple[int, ...]) –

• seed (Callable[[Array], Array]) –

• split (Callable[[Array, tuple[int, ...]], Array]) –

• random_bits (Callable[[Array, int, tuple[int, ...]], Array]) –

• fold_in (Callable[[Array, int], Array]) –

• name (str) –

• tag (str) –

Return type:

Hashable