题外话:tensorflow中,tf.contrib是一些尝试代码或者不稳定的代码,将来可能会被弃用或者删除;tf.layers中是一些已经确定了的代码,基本不会有大的改动;tf.nn是最基本的代码,级别最低,基本不会变,但要实现各种功能,需要自己组合别的api来实现。
使用tf.layers.batch_normalization(),首先是api接口:点击打开链接
注意:
训练:
1.设置training=True;
2.添加下面的代码,以保存滑动平均值,测试时会使用到;
Note: when training, the moving_mean and