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)