Pytorch 的 BatchNormaliztion 跟 Tensorflow 的差異很大

自己回答一下自己，看起來是裡面再訓練的時候不一樣，如果想在 tensorflow 上訓練時候也得到一模一樣的結果，重寫 BN 就可以了。