mindspore.ops.dstack
- mindspore.ops.dstack(inputs)[source]
Stacks tensors along the third axis.
1-D tensors \((N,)\) should be reshaped to \((1,N,1)\). 2-D tensors \((M,N)\) should be reshaped to \((M,N,1)\) before concatenation.
- Parameters
inputs (Union(List[Tensor], Tuple[Tensor])) – A sequence of tensors. The tensors must have the same shape along all but the third axis. 1-D or 2-D tensors must have the same shape.
- Returns
Stacked Tensor, will be at least 3-D. The output shape is similar to the output of numpy.dstack() function.
- Raises
TypeError – If inputs is not tuple or list.
ValueError – If inputs is empty.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> 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.]]]