mindspore.ops.column_stack

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

将多个tensor沿着水平方向维度堆叠成一个tensor。类似于 mindspore.ops.hstack()

参数:
  • tensors (Union[tuple[Tensor], list[Tensor]]) - 输入tensors。

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

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