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