jax.random.split

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

通过添加一个前导轴将 PRNG 密钥分割成 num 个新密钥。

参数:

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

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

返回:

一个 num 个新 PRNG 密钥的类数组对象。

返回类型:

Array