jax.extend.random 模組# define_prng_impl(*, key_shape, seed, split, ...) seed_with_impl(impl, seed) threefry2x32_p threefry_2x32(keypair, count) 套用 Threefry 2x32 雜湊。 threefry_prng_impl 指定 PRNG 金鑰形狀和運算。 rbg_prng_impl 指定 PRNG 金鑰形狀和運算。 unsafe_rbg_prng_impl 指定 PRNG 金鑰形狀和運算。