jax.random.PRNGKey#

jax.random.PRNGKey(seed, *, impl=None)[源代码]#

使用整数种子创建旧式 PRNG 密钥。

此函数生成旧式 PRNG 密钥，它们是 uint32 类型的数组。 更多信息，请参阅 PRNG 密钥 部分的注释。 如果可能，建议使用 jax.random.key() 代替。

生成的密钥不携带 PRNG 实现。 返回的密钥匹配可选 impl 参数给出的实现，或者由 jax_default_prng_impl 配置标志确定。 调用者必须确保在将此密钥作为参数传递给其他函数（例如 jax.random.split 和 jax.random.normal）时，设置相同的实现作为默认实现。

参数:

  • seed (int | ArrayLike) – 用作密钥值的 64 位或 32 位整数。

  • impl (PRNGSpecDesc | None | None) – 可选字符串，指定 PRNG 实现 (例如 'threefry2x32')

返回:

PRNG 密钥，可被随机函数以及 split 和 fold_in 使用。

返回类型:

Array