cupy.split#

cupy.split(ary, indices_or_sections, axis=0)[源代码]#

沿给定轴将数组分割成多个子数组。

参数:

ary (cupy.ndarray) – 要分割的数组。
indices_or_sections (int 或 整数序列) – 指示如何分割轴的值。如果它是整数，则被视为分割的数量，轴将被均匀分割。否则，整数表示要分割的索引。请注意，不允许使用设备内存上的序列。
axis (int) – 沿其分割数组的轴。

返回:

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

另请参阅