jax.random.key

jax.random.key(seed, *, impl=None)

根据整数种子创建一个伪随机数生成器 (PRNG) 密钥。

结果是一个包含密钥的标量数组，其 dtype 指示默认的 PRNG 实现，由可选的 impl 参数确定，否则，由调用此函数时的 jax_default_prng_impl 配置标志确定。

参数:

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

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

返回:

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

返回类型:

数组