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