view reply I am confused about the whitening function. Shouldn't it be as follows? whitened = (values - mean) / torch.rsqrt(var + 1e-8) But the function here has, whitened = (values - mean) * torch.rsqrt(var + 1e-8)