mindspore.ops.ExpandDims
- class mindspore.ops.ExpandDims[源代码]
对输入 input_x 在给定的轴上添加额外维度。
获取更多详情请查看
mindspore.ops.expand_dims()
。- 支持平台:
Ascend
GPU
CPU
样例:
>>> 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.]]]