megengine.functional.expand_dims# expand_dims(inp, axis)[源代码]# 在给定的 axis 前添加维度。 参数: inp (Tensor) – 输入张量。 axis (Union[int, Sequence[int]]) – 若干新轴的位置 返回类型: Tensor 返回: 输出张量。 实际案例 >>> import numpy as np >>> x = Tensor([1, 2]) >>> out = F.expand_dims(x, 0) >>> out.numpy().shape (1, 2)