megengine.functional.expand_dims

expand_dims(inp, axis)

在给定的axis前添加维度。

参数

• inp (Tensor) – 输入张量。

• axis (Union[int, Sequence[int]]) – 若干新轴的位置

返回类型

Tensor

返回

输出张量。

例如：

import numpy as np
from megengine import tensor
import megengine.functional as F

x = tensor([1, 2])
out = F.expand_dims(x, 0)
print(out.numpy().shape)

输出：

(1, 2)

megengine.functional.tile megengine.functional.squeeze