def split(ary, indices_or_sections, axis=0):
    if ary.ndim <= axis:
        raise IndexError('Axis exceeds ndim')
    size = ary.shape[axis]

    if numpy.isscalar(indices_or_sections):
        if size % indices_or_sections != 0:
            raise ValueError(
                'indices_or_sections must divide the size along the axes.\n'
                'If you want to split the array into non-equally-sized '
                'arrays, use array_split instead.')
    return array_split(ary, indices_or_sections, axis)
Splits an array into multiple sub arrays along a given axis.
dataset/ETHPy150Open pfnet/chainer/cupy/manipulation/split.py/split