Open peterbell10 opened 4 years ago
This is currently by design. indices_or_sections in cupy.split must reside on host memory, as it is needed on host side.
https://docs-cupy.chainer.org/en/stable/reference/generated/cupy.split.html
Note that the sequence on the device memory is not allowed.
I think we need to improve documentation that 0-dim arrays on the device memory are not allowed as well.
In NumPy,
np.split
will accept a 0 dimension array as the number of arrays to split into:In CuPy, the same expression fails: