cunumeric.split#

cunumeric.split(a: ndarray, indices: int | ndarray, axis: int = 0) → list[ndarray]#

Split an array into multiple sub-arrays as views into ary.

Parameters:

ary (ndarray) – Array to be divided into sub-arrays.
indices_or_sections (int or ndarray) –
If indices_or_sections is an integer, N, the array will be divided into N equal arrays along axis. If such a split is not possible, an error is raised.

If indices_or_sections is a 1-D array of sorted integers, the entries indicate where along axis the array is split. For example, [2, 3] would, for axis=0, result in
- ary[:2]
- ary[2:3]
- ary[3:]
If an index exceeds the dimension of the array along axis, an empty sub-array is returned correspondingly.
axis (int, optional) – The axis along which to split, default is 0.

Returns:

sub-arrays – A list of sub-arrays as views into ary.

Return type:

list[ndarray]

Raises:

ValueError – If indices_or_sections is given as an integer, but a split does not result in equal division.