jax.random.PRNGKey

内容

jax.random.PRNGKey#

jax.random.PRNGKey(seed, *, impl=None)[source]#

使用整数种子创建旧版 PRNG 键。

此函数生成旧式旧版 PRNG 键，它们是 dtype 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 使用。

返回类型:

KeyArray

目录