megengine.functional.nn.adaptive_avg_pool2d

adaptive_avg_pool2d(inp, oshp)

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

更多信息参见 AvgPool2d。

参数

• inp (Tensor) – 输入张量。

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

返回类型

Tensor

返回

输出张量。

上一页

megengine.functional.nn.max_pool2d

下一页

megengine.functional.nn.adaptive_max_pool2d