cupy.dsplit

cupy.dsplit(ary, indices_or_sections)

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

这等价于使用 axis=2 的 split 函数。

另见

cupy.split() 获取更多详细信息，numpy.dsplit()