jax.random.PRNGKey#
- jax.random.PRNGKey(seed, *, impl=None)[源代码][源代码]#
给定一个整数种子,创建一个遗留的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) – 可选字符串,指定PRNG实现(例如
'threefry2x32'
)
- 返回:
一个PRNG键,可被随机函数以及
split
和fold_in
使用。- 返回类型:
KeyArray