Pytorch 自适应平均池化算子 “nn.AdaptiveAvgPool2d” 计算逻辑解析

目前对 pytorch nn.AdaptiveAvgPool2d 算子普遍存在的复现方式仅合其形，不对其意，没有真正复现其功能。本文通过对AdaptiveAvgPool2d 的计算逻辑进行分析，完成了通用模式下对 NCHW 模式数据的自适应平均 2 维池化操作，并可以此为基础迁移至其他 AI 框架中。

查看全文