jax.extend.random.threefry_2x32

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

应用 Threefry 2x32 哈希。

参数:

• keypair – 用于键的一对32位无符号整数。

• count – 用于计数的 dtype uint32 数组。

返回:

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