cupy.vsplit

cupy.vsplit(ary, indices_or_sections)

沿第一个轴将数组分割成多个子数组。

这等同于使用 axis=0 的 split 函数。

另请参阅

cupy.split() 了解更多详情，numpy.dsplit()