jax.random.fold_in#

jax.random.fold_in(key, data)[source]#

将数据融入 PRNG 密钥以形成一个新的 PRNG 密钥。

参数:
  • key (ArrayLike) – 一个 PRNG 密钥(来自 keysplitfold_in)。

  • data (IntegerArray) – 一个 32 位整数,表示要融入到密钥中的数据。

返回:

一个新的 PRNG 密钥,它是输入的确定性函数,并且在生成新的伪随机值流方面是统计安全的。

返回类型:

Array