mlx.core.split

split(a: array, /, indices_or_sections: int | Sequence[int], axis: int = 0, *, stream: None | Stream | Device = None) → array

沿给定轴分割数组。

参数:

• a (array) – 输入数组。

• indices_or_sections (int or list(int)) – 如果 indices_or_sections 是整数，则将数组分割成指定数量的相等大小的部分。如果不可能，则会引发错误。如果 indices_or_sections 是列表，则列表包含沿给定轴的每个子数组的起始索引。

• axis (int, optional) – 沿此轴进行分割，默认为 0。

返回:

分割后的数组列表。

返回类型:

list(array)