mindspore.ops.hstack

查看源文件
mindspore.ops.hstack(tensors)[源代码]

将多个tensor沿着水平方向堆叠。

说明

  • graph mode 下,不支持 mindspore.float64 类型的8-D tensor的动态rank输入。

  • 对于一维tensor,沿第一个轴堆叠。其他维度的tensor沿第二个轴堆叠。

  • 对于维度大于一的tensor,除了第二个轴外,所有tensor的shape必须相同。对于一维tensor,可拥有任意的长度。

参数:
  • tensors (Union[tuple[Tensor], list[Tensor]]) - 由多个tensor组成的tuple或list。

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

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