mindspore.ops.Stack
- class mindspore.ops.Stack(axis=0)[source]
Stacks a list of tensors in specified axis.
Refer to
mindspore.ops.stack()
for more details.- Parameters
axis (int, optional) – Dimension to stack. The range is [-(R+1), R+1). Default:
0
.
- Inputs:
input_x (Union[tuple, list]) - A Tuple or list of Tensor objects with the same shape and type.
- Outputs:
Tensor. A stacked Tensor with the same type as input_x.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import mindspore >>> from mindspore import Tensor, ops >>> import numpy as np >>> data1 = Tensor(np.array([0, 1]).astype(np.float32)) >>> data2 = Tensor(np.array([2, 3]).astype(np.float32)) >>> stack = ops.Stack() >>> output = stack([data1, data2]) >>> print(output) [[0. 1.] [2. 3.]]