sakura.utils.kl_divergence

KL divergence for loss computation

Classes

KLDivergence

Computes KL Divergence between encoded samples and target distribution Assumes both distributions are multivariate Gaussian with diagonal covariance