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]]