jax.extend.random.define_prng_impl

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

参数：

• key_shape (元组[整数, ...])

• seed (可调用对象[[数组], 数组])

• split (可调用对象[[Array, 元组[int, ...]], Array])

• random_bits (可调用对象[[Array, int, 元组[int, ...]], Array])

• fold_in (可调用对象[[Array, int], Array])

• name (字符串)

• tag (字符串)

返回类型:

可哈希