jax.extend.random.rbg_prng_impl

jax.extend.random.rbg_prng_impl = ((4,), <function _rbg_seed>, <function _rbg_split>, <function _rbg_random_bits>, <function _rbg_fold_in>, 'rbg', 'rbg')

指定 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 函式建立。