megengine.functional.squeeze

squeeze(inp, axis=None)

删除形状(shape)中下标为1的维度。

参数

• inp (Tensor) – 输入张量。

• axis (Union[int, Sequence[int], None]) – 将要被移除的轴的位置。

返回类型

Tensor

返回

输出张量。

实际案例

>>> import numpy as np
>>> x = Tensor(np.array([1, 2], dtype=np.int32).reshape(1, 1, 2, 1))
>>> out = F.squeeze(x, 3)
>>> out.numpy().shape
(1, 1, 2)

上一页

megengine.functional.expand_dims

下一页

megengine.functional.concat