jax.random.fold_in

内容

jax.random.fold_in#

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

将数据折叠到 PRNG 密钥中以形成新的 PRNG 密钥。

参数:
  • key (KeyArrayLike) – PRNG 密钥(来自 keysplitfold_in)。

  • data (IntegerArray) – 表示要折叠到密钥中的数据的 32 位整数。

返回值:

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

返回类型:

KeyArray