megengine.functional.nn.adaptive_avg_pool2d¶

adaptive_avg_pool2d(inp, oshp)[源代码]¶

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

更多信息参见 AvgPool2d。

参数

inp (Tensor) – 输入张量。
oshp (Union[Tuple[int, int], int, Tensor]) – OH, OW)` size of the output shape.

返回类型

Tensor

返回

输出张量。

megengine.functional.nn.max_pool2d

megengine.functional.nn.adaptive_max_pool2d