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)

megengine.functional.broadcast_to

megengine.functional.squeeze