AdaptiveAvgPool2d

In [1]: import torch In [2]: a = torch.rand([4,3,4,4]) In [3]: a.size() Out[3]: torch.Size([4, 3, 4, 4]) In [4]: b = torch.nn.functional.adaptive_avg_pool2d(a, (1,1)) # 自适应池化，指定池化输出尺寸为 1 * 1 In [

>>阅读原文<<