mindspore.ops.vstack

查看源文件
mindspore.ops.vstack(inputs)[源代码]

将多个tensor沿着垂直方向进行堆叠。

等同于将输入按第一个轴进行拼接。 一维tensor (N,) 重新排列为 (1,N) ,然后按第一个轴进行拼接。

参数:
  • inputs (Union(List[tensor], Tuple[tensor])) - 一维或二维输入。

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> x1 = mindspore.tensor([1, 2, 3])
>>> x2 = mindspore.tensor([4, 5, 6])
>>> mindspore.ops.vstack((x1, x2))
Tensor(shape=[2, 3], dtype=Int64, value=
[[1, 2, 3],
 [4, 5, 6]])
>>> x1 = mindspore.tensor([[1],[2],[3]])
>>> x2 = mindspore.tensor([[4],[5],[6]])
>>> mindspore.ops.vstack([x1, x2])
Tensor(shape=[6, 1], dtype=Int64, value=
[[1],
 [2],
 [3],
 [4],
 [5],
 [6]])