jax.random.key

内容

jax.random.key#

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

使用整数种子创建伪随机数生成器 (PRNG) 键。

结果是一个包含键的标量数组，其数据类型指示默认的 PRNG 实现，该实现由可选的 impl 参数确定，或者在没有 impl 参数的情况下，由调用此函数时的 jax_default_prng_impl 配置标志确定。

参数：:

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

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

返回：:

一个标量 PRNG 键数组，可供随机函数以及 split 和 fold_in 使用。

返回类型：:

KeyArray

内容