megengine.functional.squeeze

squeeze(inp, axis=None)[源代码]

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

参数
  • inp – 输入张量。

  • axis – 将要被移除的轴的位置。

返回

输出张量。

例如:

import numpy as np
from megengine import tensor
import megengine.functional as F

x = tensor(np.array([1, 2], dtype=np.int32).reshape(1, 1, 2, 1))
out = F.squeeze(x, 3)
print(out.numpy().shape)

输出:

(1, 1, 2)