jax.random.fold_in

jax.random.fold_in(key, data)

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

参数:

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

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

返回:

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

返回类型:

Array