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 使用。

返回类型:

Array