megengine.functional.nn.adaptive_avg_pool2d

adaptive_avg_pool2d(inp, oshp)

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

更多信息参见 AvgPool2d。

参数:

• inp (Tensor) – 输入张量。

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

返回类型:

Tensor

返回:

输出张量。