jax.random.split

jax.random.split(key, num=2)

通过添加一个前导轴，将一个 PRNG 键拆分为 num 个新键。

参数:

• key (KeyArrayLike) – 一个PRNG密钥（来自 key、split、fold_in）。

• num (int | tuple[int, ...]) – 可选，一个正整数（或整数元组），表示要生成的键的数量（或形状）。默认为2。

返回:

一个类似数组的 num 个新的 PRNG 键对象。

返回类型:

KeyArray