text: Title: KL Loss.
  Body: Hi, I just noticed that the KL Loss in the VAE paper would look like this: 
  0.5 * torch.sum(torch.exp(logVar) + mean ** 2 - 1. - logVar)
  
  And here, the KL Loss is:
  torch.mean(0.5 * torch.sum(torch.exp(logVar) + mean ** 2 - 1. - logVar, 1))
  
  What's your thought in this? 