tools模块中的batchnorm计算

batch Normalization的计算时，均值和方差的计算在您的函数中是基于batch这一维的，但是我认为应该是取[batch, height, width]这三维的均值和方差