cupy.array_split#

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

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

此函数几乎等同于 cupy.split()。唯一的区别是此函数允许整数部分不均匀地分割轴。

参见

cupy.split() 了解更多详情, numpy.array_split()