Pytorch Batch Normalization Formula