前言
本文以 2D 图像数据为例对 BN(Batch Normalization), LN(Layer Normalization), IN(Instance Normalization), WN(Weight Normalization) 的 Pytorch 源码进行了分析解读。以期通过对源码的分析可以加深对这几种方法的理解。
我将首先简单分析一下 Pytorch 中对几种方法的 Python 实现,虽然深入探究一下底层 C++ 源码的实现,最后根据我们学习到的方法复现出 Pytorch 的实现效果。