mindspore.ops.expand_dims

mindspore.ops.expand_dims(input_x, axis)

为输入tensor新增额外的轴。

说明

• input_x 的维度应该大于等于1。

• 如果指定的 axis 是负数，那么它会从后往前，从1开始计算index。

参数：

• input_x (Tensor) - 输入tensor。

• axis (int) - 新增的轴。仅接受常量。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input_tensor = mindspore.tensor([[2, 2], [2, 2]], mindspore.float32)
>>> output = mindspore.ops.expand_dims(input_tensor, 0)
>>> print(output)
[[[2. 2.]
  [2. 2.]]]