cupy.split

cupy.split(ary, indices_or_sections, axis=0)

沿给定轴将数组拆分为多个子数组。

参数:

• ary (cupy.ndarray) – 要分割的数组。

• indices_or_sections (int or sequence of ints) – 一个指示如何划分轴的值。如果它是一个整数，则被视为段数，并且轴被均匀划分。否则，整数表示要分割的索引。请注意，设备内存中的序列是不允许的。

• axis (int) – 沿其分割数组的轴。

返回:

子数组列表。每个数组是对应输入数组的视图。

参见

numpy.split()