megengine.functional.expand_dims#

expand_dims(inp, axis)[源代码]#

在给定的 axis 前添加维度。

参数:
返回类型:

Tensor

返回:

输出张量。

实际案例

>>> import numpy as np
>>> x = Tensor([1, 2])
>>> out = F.expand_dims(x, 0)
>>> out.numpy().shape
(1, 2)