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) – 可选字符串，指定PRNG实现（例如
'threefry2x32'）
- 返回:
一个标量 PRNG 键数组，可被随机函数以及
split和fold_in使用。- 返回类型:
KeyArray