The VAE maximizes a lower bound on the log-likelihood of the data. The bound is the reconstruction term minus a KL penalty, and it can be estimated with samples from the posterior.
Implement elbo(log_px_given_z, kl), returning the reconstruction term minus the KL.
Subtract the KL from the reconstruction log-likelihood.
No results yet
Run your code to check it against the examples, or submit to run every test.