jax.extend.random.threefry_prng_impl

jax.extend.random.threefry_prng_impl = ((2,), <function threefry_seed>, <function threefry_split>, <function threefry_random_bits>, <function threefry_fold_in>, 'threefry2x32', 'fry')

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