mx.nd.stack

Description

Join a sequence of arrays along a new axis. The axis parameter specifies the index of the new axis in the dimensions of the result. For example, if axis=0 it will be the first dimension and if axis=-1 it will be the last dimension.

Example:

x = [1, 2]
y = [3, 4]
stack(x, y) = [[1, 2],
[3, 4]]
stack(x, y, axis=1) = [[1, 3],
[2, 4]]

**Example**::

x = [1, 2]
y = [3, 4]
stack(x, y) = [[1, 2],
[3, 4]]
stack(x, y, axis=1) = [[1, 3],
[2, 4]]

Arguments

Argument

Description

data

NDArray-or-Symbol[].

List of arrays to stack

axis

int, optional, default=’0’.

The axis in the result array along which the input arrays are stacked.

num.args

int, required.

Number of inputs to be stacked.

Value

out The result mx.ndarray