mindspore.ops.dstack
- mindspore.ops.dstack(inputs)[源代码]
将多个Tensor沿着第三维度进行堆叠。
1-D Tensor \((N,)\) 重新排列为 \((1,N,1)\) ,2-D Tensor \((M,N)\) 重新排列为 \((M,N,1)\) 。
- 参数:
inputs (Union(List[Tensor], Tuple[Tensor])) - 一个Tensor序列。除了第三个轴外,所有的 Tensor必须有相同的shape。如果是1-D或2-D的Tensor,则它们的shape必须相同。
- 返回:
堆叠后的Tensor,其维度至少为3。输出shape与 numpy.dstack() 类似。
- 异常:
TypeError - 如果 inputs 不是list或tuple。
ValueError - 如果 inputs 为空。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import numpy as np >>> from mindspore import Tensor, ops >>> x1 = Tensor(np.arange(1, 7).reshape(2, 3)) >>> x2 = Tensor(np.arange(7, 13).reshape(2, 3)) >>> out = ops.dstack([x1, x2]) >>> print(out.asnumpy()) [[[ 1. 7.] [ 2. 8.] [ 3. 9.]] [[ 4. 10.] [ 5. 11.] [ 6. 12.]]]