diffusion_models.losses.kl_divergence

Functions

gaussian_kl(p_mean, p_var, q_mean, q_var)	Calculate KL Divergence of 2 Gaussian distributions.
log_gaussian_kl(p_mean, p_logvar, q_mean, ...)	Calculate KL Divergence of 2 Gaussian distributions.