mindspore.Tensor.expand_dims

mindspore.Tensor.expand_dims(axis)[源代码]

沿指定轴扩展Tensor维度。

参数:
  • axis (int) - 扩展维度指定的轴。

返回:

Tensor,指定轴上扩展的维度为1。

异常:
  • TypeError - axis不是int类型。

  • ValueError - axis的取值不在[-self.ndim - 1, self.ndim + 1)范围内。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((2,2), dtype=np.float32))
>>> print(x)
[[1. 1.]
[1. 1.]]
>>> print(x.shape)
(2, 2)
>>> y = x.expand_dims(axis=0)
>>> print(y)
[[[1. 1.]
[1. 1.]]]
>>> print(y.shape)
(1, 2, 2)