mindspore.ops.hstack

mindspore.ops.hstack(tensors)[源代码]

将多个Tensor沿着水平方向进行堆叠。 对于1-D Tensor,沿第一个轴进行堆叠。其他维度的Tensor沿第二个轴进行堆叠。

参数:
  • tensors (Union[tuple[Tensor], list[Tensor]]) - 包含多个Tensor。对于维度大于1-D的Tensor,除了第二个轴外,所有的 Tensor必须有相同的shape。对于1-D Tensor,可拥有任意的长度。

返回:

堆叠后的Tensor。

异常:
  • TypeError - 如果 tensors 不是 list或tuple。

  • TypeError - 如果 tensors 的元素不是 Tensor。

  • ValueError - 如果 tensors 为空。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor, ops
>>> x1 = Tensor([1, 1, 1])
>>> x2 = Tensor([2, 2, 2])
>>> output = ops.hstack((x1, x2))
>>> print(output)
[1. 1. 1. 2. 2. 2.]