mindspore.ops.column_stack

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

将多个Tensor沿着水平方向维度堆叠成一个Tensor,即按列拼接。Tensor其他维度拼接的结果 维度不变。类似于 mindspore.ops.hstack()

参数:
  • tensors (Union[tuple[Tensor], list[Tensor]]) - 包含多个Tensor。除了需要拼接的轴外,所有的 Tensors必须有相同的shape。

返回:

将输入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.column_stack((x1, x2))
>>> print(output)
[[1 2]
 [1 2]
 [1 2]]