jax.extend.random.threefry_2x32

jax.extend.random.threefry_2x32(keypair, count)

应用 Threefry 2x32 哈希。

参数:

• keypair – 用于密钥的 32 位无符号整数对。

• count – 一个 dtype 为 uint32 的数组，用于计数。

返回:

一个 dtype 为 uint32 的数组，其形状与 count 相同。