All Questions
Tagged with pytorch variational-autoencoder
1 question
7
votes
2
answers
3k
views
How is this Pytorch expression equivalent to the KL divergence?
I found the following PyTorch code (from this link)
-0.5 * torch.sum(1 + sigma - mu.pow(2) - sigma.exp())
where mu is the mean ...