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
>>> import numpy as np
>>> x1 = mindspore.tensor(np.arange(1, 7).reshape(2, 3))
>>> x2 = mindspore.tensor(np.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.]]]