AdaptiveAvgPool2d 差异对比

torch.nn.AdaptiveAvgPool2d

torch.nn.AdaptiveAvgPool2d(
   output_size
)

更多请查看 torch.nn.AdaptiveAvgPool2d.

megengine.module.AdaptiveAvgPool2d

megengine.module.AdaptiveAvgPool2d(
    oshp
    ** kwargs

)

更多请查看 megengine.module.AdaptiveAvgPool2d.

使用差异

输入张量的形状

PyTorch 支持 NCHW 或者 CHW 的输入，MegEngine 支持 NCHW 的输入。