chainerx.vsplit¶
-
chainerx.
vsplit
(ary, indices_or_sections)¶ Splits an array into multiple sub-arrays vertically (row-wise).
- Parameters
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.
- Returns
A list of sub arrays. Each array is a partial view of the input array.
- Return type
list of
ndarray
s
Note
During backpropagation, this function propagates the gradients of the output arrays to the input array
ary
.See also