dpnp.dsplit

dpnp.dsplit(ary, indices_or_sections)[source]

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

Please refer to the dpnp.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.

For full documentation refer to numpy.dsplit.

Parameters:

ary ({dpnp.ndarray, usm_ndarray}) -- Array to be divided into sub-arrays.
indices_or_sections ({int, sequence of ints}) -- If indices_or_sections is an integer, N, the array will be divided into N equal arrays along the third axis. If such a split is not possible, an error is raised. If indices_or_sections is a sequence of sorted integers, the entries indicate where along the third axis the array is split.

Returns:

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

Return type:

list of dpnp.ndarray

See also

dpnp.split: Split array into multiple sub-arrays of equal size.

Examples

>>> import dpnp as np
>>> x = np.arange(16.0).reshape(2, 2, 4)
>>> x
array([[[ 0.,  1.,  2.,  3.],
        [ 4.,  5.,  6.,  7.]],
       [[ 8.,  9., 10., 11.],
        [12., 13., 14., 15.]]])
>>> np.dsplit(x, 2)
[array([[[ 0.,  1.],
         [ 4.,  5.]],
        [[ 8.,  9.],
         [12., 13.]]]),
 array([[[ 2.,  3.],
         [ 6.,  7.]],
        [[10., 11.],
         [14., 15.]]])]
>>> np.dsplit(x, np.array([3, 6]))
[array([[[ 0.,  1.,  2.],
         [ 4.,  5.,  6.]],
        [[ 8.,  9., 10.],
         [12., 13., 14.]]]),
 array([[[ 3.],
         [ 7.]],
        [[11.],
         [15.]]]),
 array([])]