jax.extend.random.unsafe_rbg_prng_impl

jax.extend.random.unsafe_rbg_prng_impl = ((4,), <function _rbg_seed>, <function _unsafe_rbg_split>, <function _rbg_random_bits>, <function _unsafe_rbg_fold_in>, 'unsafe_rbg', 'urbg')

指定 PRNG 金鑰形狀和運算。

PRNG 實作由金鑰型別 K 和一組對此類金鑰進行運算的函式決定。金鑰型別 K 是一種陣列型別，其元素型別為 uint32，形狀由 key_shape 指定。每個運算的型別簽名為

seed :: int[] -> K
fold_in :: K -> int[] -> K
split[shape] :: K -> K[*shape]
random_bits[shape, bit_width] :: K -> uint<bit_width>[*shape]

PRNG 實作透過 PRNGKeyArray 類別適應金鑰 K 的類陣列物件，該類別應透過 random_seed 函式建立。