在实现NiN的时候,实现GAP使用了nn.AdaptiveAvgPool2d()函数,这里总结一下这两个函数的使用。
nn.AdaptiveAvgPool2d()的帮助文档nn.AdaptiveAvgPool2d(output_size: Union[int, Tuple[int, ...]]