mindspore.numpy.column_stack

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

将一维Tensor作为列堆叠为二维Tensor。二维Tensor保持原样堆叠,如同 np.hstack

参数:
  • tup (Union[Tensor, tuple, list]) - 一组一维或二维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.column_stack((x1, x2))
>>> print(output)
[[1 4]
 [2 5]
 [3 6]]