Splits an array into multiple sub arrays along a given axis.
This function is almost equivalent to cupy.split(). The only
difference is that this function allows an integer sections that does not
evenly divide the axis.
cupy.split() for more detail, numpy.array_split()