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.broadcast_to

megengine.functional.squeeze