Split array into multiple sub-arrays along the 3rd axis (depth).
ary (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 a sequence on the device memory is not allowed.
A list of sub arrays. Each array is a partial view of the input array.
- Return type
During backpropagation, this function propagates the gradients of the output arrays to the input array