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) – 可选字符串，指定PRNG实现（例如 'threefry2x32'）

返回:

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

返回类型:

数组