mindspore.numpy.expand_dims

mindspore.numpy.expand_dims(a, axis)

扩展Tensor的shape。在扩展后的Tensor的shape中插入一个新轴，该轴将出现在指定的位置。

参数：

• a (Tensor) - 输入Tensor数组。

• axis (Union[int, list(int), tuple(int)]) - 新轴在扩展轴中的位置。

返回：

Tensor，指定轴位置的维度数增加后的Tensor。

异常：

• TypeError - 如果输入参数非上述给定的类型。

• ValueError - 如果 axis 超出了 a.ndim 。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> x = np.ones((2,2))
>>> x = np.expand_dims(x,0)
>>> print(x.shape)
(1, 2, 2)