mindspore.ops.dstack

mindspore.ops.dstack(tensors)[source]

Stacks tensors along the third axis.

Note

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.
The tensors must have the same shape along all but the third axis. 1-D or 2-D tensors must have the same shape.

Parameters: tensors (Union(List[Tensor], Tuple[Tensor])) – The list of tensors or tuple of tensors.
Returns: Tensor

Supported Platforms:: Ascend GPU CPU

Examples

>>> import mindspore
>>> x1 = mindspore.tensor(mindspore.ops.arange(1, 7).reshape(2, 3))
>>> x2 = mindspore.tensor(mindspore.ops.arange(7, 13).reshape(2, 3))
>>> out = mindspore.ops.dstack([x1, x2])
>>> print(out.asnumpy())
[[[ 1.  7.]
  [ 2.  8.]
  [ 3.  9.]]
 [[ 4. 10.]
  [ 5. 11.]
  [ 6. 12.]]]