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 ndarrays

Note

During backpropagation, this function propagates the gradients of the output arrays to the input array ary.

See also

numpy.vsplit()