megengine.functional.adaptive_avg_pool2d

adaptive_avg_pool2d(inp, oshp)

对输入进行二维平均池化。

更多信息参见 AvgPool2d。

参数

• inp (Tensor) – 输入张量。

• oshp (Union[Tuple[int, int], int, Tensor]) – 输出图像的大小 (OH, OW)。

返回类型

Tensor

返回

输出张量。

megengine.functional.max_pool2d megengine.functional.adaptive_max_pool2d