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)