cunumeric.dsplit#

cunumeric.dsplit(a: ndarray, indices: int | ndarray) list[ndarray]#

Split array into multiple sub-arrays along the 3rd axis (depth).

Please refer to the split documentation. dsplit is equivalent to split with axis=2, the array is always split along the third axis provided the array dimension is greater than or equal to 3.

See also

numpy.dsplit

Availability:

Multiple GPUs, Multiple CPUs