mindspore.ops.stack

mindspore.ops.stack(tensors, axis=0)

在指定轴上对输入的tensors进行堆叠。

参数：

• tensors (Union[tuple, list]) - 输入tensors。

• axis (int) - 指定轴，默认 0 。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input_x1 = mindspore.tensor([0, 1], mindspore.float32)
>>> input_x2 = mindspore.tensor([2, 3], mindspore.float32)
>>> output = mindspore.ops.stack((input_x1, input_x2), 0)
>>> print(output)
[[0. 1.]
 [2. 3.]]