mlx.core.random.split

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

Split a PRNG key into sub keys.

參數:

• key (array) -- Input key to split.

• num (int, optional) -- Number of sub keys. Default: 2.

回傳:

The array of sub keys with num as its first dimension.

回傳型別:

array