cupy.split#

cupy.split(ary, indices_or_sections, axis=0)[source]#

Splits an array into multiple sub arrays along a given axis.

Parameters:
  • ary (cupy.ndarray) – Array to split.

  • indices_or_sections (int or sequence of ints) – A value indicating how to divide the axis. If it is an integer, then is treated as the number of sections, and the axis is evenly divided. Otherwise, the integers indicate indices to split at. Note that the sequence on the device memory is not allowed.

  • axis (int) – Axis along which the array is split.

Returns:

A list of sub arrays. Each array is a view of the corresponding input array.

See also

numpy.split()