megengine.functional.nn.max_pool2d

max_pool2d(inp, kernel_size, stride=None, padding=0)[源代码]

对输入张量进行二维最大池化。

更多信息参见 MaxPool2d。

参数

  • 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

返回类型

Tensor

返回

输出张量。

上一页

megengine.functional.nn.avg_pool2d

下一页

megengine.functional.nn.adaptive_avg_pool2d