mindspore.ops.ExpandDims
- class mindspore.ops.ExpandDims[source]
Adds an additional dimension to input_x at the given axis.
Refer to
mindspore.ops.expand_dims()
for more details.- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> input_tensor = Tensor(np.array([[2, 2], [2, 2]]), mindspore.float32) >>> expand_dims = ops.ExpandDims() >>> output = expand_dims(input_tensor, 0) >>> print(output) [[[2. 2.] [2. 2.]]]