10f7e5ffaa583d09912e72e82353fe64814e7967,gpytorch/mlls/variational_elbo.py,VariationalELBO,forward,#VariationalELBO#Any#Any#,25

Before Change

        log_likelihood = self.likelihood.variational_log_probability(variational_dist_f, target, **kwargs).div(
            num_batch
        )
        kl_divergence = torch.distributions.kl.kl_divergence(variational_dist_u, prior_dist).sum().div(self.num_data)

        if self.combine_terms:
            res = log_likelihood - kl_divergence

After Change

        kl_divergence = torch.distributions.kl.kl_divergence(variational_dist_u, prior_dist)

        if log_likelihood.numel() == 1:
            kl_divergence = kl_divergence.sum()

        kl_divergence = kl_divergence.div(self.num_data)

        if self.combine_terms:
            res = log_likelihood - kl_divergence

Instances