mindspore.ops.vstack
- mindspore.ops.vstack(inputs)[源代码]
将多个Tensor沿着竖直方向进行堆叠。
相当于将输入沿着第一个轴进行拼接。 1-D Tensor \((N,)\) 重新排列为 \((1, N)\) ,然后沿着第一个轴进行拼接。
- 参数:
inputs (Union(List[tensor], Tuple[tensor])) - 一个1-D或2-D Tensor序列。除了第一个轴外,所有的 Tensor必须有相同的shape。如果是1-DTensor,则它们的shape必须相同。
- 返回:
堆叠后的Tensor,其维度至少为3。输出shape与 numpy.vstack() 类似。
- 异常:
TypeError - 如果 inputs 不是list或tuple。
ValueError - 如果 inputs 为空。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore.numpy as np >>> x1 = np.array([3, 1, 4]) >>> x2 = np.array([1, 5, 9]) >>> out = ops.vstack([x1, x2]) >>> print(out) [[3 1 4] [1 5 9]]