split(ary, indices_or_sections, axis=0)¶
Splits an array into multiple sub arrays along a given axis.
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.
axis (int) – Axis along which the array is split.
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