mindspore.ops.dstack

mindspore.ops.dstack(tensors)[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

tensors (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 tensors is not tuple or list.
ValueError – If tensors 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.]]]