mindspore.ops.hstack
- mindspore.ops.hstack(tensors)[源代码]
将多个Tensor沿着水平方向进行堆叠。 对于1-D Tensor,沿第一个轴进行堆叠。其他维度的Tensor沿第二个轴进行堆叠。
- 参数:
tensors (Union[tuple[Tensor], list[Tensor]]) - 包含多个Tensor。对于维度大于1-D的Tensor,除了第二个轴外,所有的 Tensor必须有相同的shape。对于1-D 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.hstack((x1, x2)) >>> print(output) [1. 1. 1. 2. 2. 2.]