mlx.core.random.split

split(key: array, num: int = 2, stream: None | Stream | Device = None) → array

将 PRNG 密钥拆分为子密钥。

参数:

• key (数组) – 要拆分的输入密钥。

• num (int, 可选) – 子密钥的数量。默认为：2。

返回值:

子密钥的数组，其中 num 作为其第一个维度。

返回类型:

数组