该部分代码基于MNIST手写数据集构造了一个四层的全连接层神经网络。通过改变不同参数来测试BN对于模型性能的影响。
- Udacity Deep Learning Course
- https://www.zhihu.com/question/38102762
- https://r2rt.com/implementing-batch-normalization-in-tensorflow.html
- Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift