jax.random.fold_in

jax.random.fold_in(key, data)

将数据折叠到 PRNG 密钥中，形成一个新的 PRNG 密钥。

参数:

• key (ArrayLike) – 一个 PRNG 密钥（来自 key、split、fold_in）。

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

返回:

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

返回类型:

Array