megengine.functional.avg_pool2d

avg_pool2d(inp, kernel_size, stride=None, padding=0, mode='AVERAGE_COUNT_EXCLUDE_PADDING')[源代码]

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

更多信息参见 AvgPool2d。

参数

  • inp (Tensor) – 输入张量。

  • kernel_size (Union[int, Tuple[int, int]]) – 窗口的尺寸。

  • stride (Union[int, Tuple[int, int], None]) – 窗口的步长。 如果该值没有给出，则将值设置为 kernel_size 。默认： None

  • padding (Union[int, Tuple[int, int]]) – 添加在每一侧的隐含零值填充。 默认：0

  • mode (str) – 是否把填充值计算在内。默认：”AVERAGE_COUNT_EXCLUDE_PADDING”

返回类型

Tensor

返回

输出张量。

megengine.functional.deformable_conv2d megengine.functional.max_pool2d