cupy.array_split#

cupy.array_split(ary, indices_or_sections, axis=0)[source]#

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

此函数与 cupy.split() 几乎等效。唯一的区别是,此函数允许对轴进行不等分的整数切片分割。

另请参阅

cupy.split() 以获取更多详情,numpy.array_split()