chainerx.stack

chainerx.stack(arrays, axis=0)

Stacks arrays along a new axis.

Parameters

  • arrays (sequence of ndarrays) – Arrays to be stacked.

  • axis (int) – Axis along which the arrays are stacked.

Returns

Stacked array.

Return type

ndarray

Note

During backpropagation, this function propagates the gradient of the output array to the input arrays in arrays.

See also

numpy.stack()