Where does `log_var` and `mean_mu` come from in `kl_loss`?

[ijkilchenko]

The scope of those two variables doesn't seem right to me.