mindspore.numpy.hstack

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

按顺序水平堆叠Tensor。此操作等效于沿第二轴连接Tensor,但当输入为一维Tensor时,则沿第一轴连接。

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

返回:

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

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

  • ValueError - 如果 tup 为空。

支持平台:

Ascend GPU CPU

样例:

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