mindspore.ops.ExpandDims

class mindspore.ops.ExpandDims

对输入 input_x 在给定的轴上添加额外维度。

获取更多详情请查看 mindspore.ops.expand_dims() 。

输入：

• input_x (Tensor) - Tensor的shape是 \((x_1, x_2, ..., x_R)\) 。

• axis (int) - 指定需添加额外维度的轴。轴值必须在[-input_x.ndim-1, input_x.ndim]范围内。只支持常量值。

输出：

Tensor，Tensor的shape为 \((1, x_1, x_2, ..., x_R)\) （ axis 的值为0）。其数据类型与 input_x 的相同。

支持平台：

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.]]]