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键的形状和操作。