megengine.functional.nn.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) – whether to count padding values, set to “average” will do counting. Default: “average_count_exclude_padding”

返回类型

Tensor

返回

输出张量。

上一页

megengine.functional.nn.sliding_window_transpose

下一页

megengine.functional.nn.max_pool2d