mindspore.ops.vstack
- mindspore.ops.vstack(inputs)[源代码]
将多个tensor沿着垂直方向进行堆叠。
等同于将输入按第一个轴进行拼接。 一维tensor
重新排列为 ,然后按第一个轴进行拼接。- 参数:
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]])