When I searched KL-divergence between two Gaussians, I got this which is diffenrent from your KL loss https://stats.stackexchange.com/questions/7440/kl-divergence-between-two-univariate-gaussians 
When I searched KL-divergence between two Gaussians, I got this which is diffenrent from your KL loss

https://stats.stackexchange.com/questions/7440/kl-divergence-between-two-univariate-gaussians