mindspore.numpy.vstack

查看源文件
mindspore.numpy.vstack(tup)[源代码]

按顺序垂直堆叠Tensor。此操作等同于沿第一轴进行连接。对于一维Tensor,应首先将其shape变为 (1, N) ,然后沿第一轴进行连接。

参数:
  • tup (Union[Tensor, tuple, list]) - 一组一维或二维Tensor。Tensor必须在除第一轴之外的所有轴上具有相同的shape。一维Tensor必须具有相同的shape。

返回:

堆叠后的Tensor,由给定的Tensor堆叠而成。

异常:
  • TypeError - 如果 tup 不是Tensor、list或tuple。

  • ValueError - 如果 tup 为空。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore.numpy as np
>>> x1 = np.array([1, 2, 3]).astype('int32')
>>> x2 = np.array([4, 5, 6]).astype('int32')
>>> output = np.vstack((x1, x2))
>>> print(output)
[[1 2 3]
 [4 5 6]]