【pytorch学习】关于torch.nn.MaxPool2d和torch.nn.functional.max_pool2d

两者之间的区别与联系

首先给出结论，torch.nn.MaxPool2d和torch.nn.functional.max_pool2d两者本质上是一样的。
具体可以参考torch.nn.MaxPool2d的源代码，核心源代码如下所示：

from .. import functional as F

class MaxPool2d(_MaxPoolNd):
    kernel_size: _size_2_t
    stride: _size_2_t
    padding: _size_2_t
    dilation: _size_2_t

    def forward(self, input: Tensor) -> Tensor:
        return F.max_pool2d(input, self.kernel_size, self.stride,
                            self.padding, self.dilation, self.ceil_mode,
                            self.return_indices)

由此可见，torch.nn.MaxPool2d在自己的forward()方法中调用了torch.nn.functional.max_pool2d。
而pytorch官网在介绍torch.nn.functional.max_pool2d时，也是直接说

See MaxPool2d for details.

因此两者本质上是一样的。

至于为什么同一个任务定义两种方法，可能是为了符合不同人使用PyTorch的编码风格。

资料参考

https://pytorch.org/docs/stable/generated/torch.nn.functional.max_pool2d.html#torch.nn.functional.max_pool2d

https://stackoverflow.com/questions/58514197/difference-between-nn-maxpool2d-vs-nn-functional-max-pool2d?noredirect=1