4275d25af066b667a0787f9639b652789f9a0338,gpytorch/likelihoods/multitask_gaussian_likelihood.py,MultitaskGaussianLikelihood,forward,#MultitaskGaussianLikelihood#Any#,57

Before Change

            task_var_lv = RootLazyTensor(self.task_noise_covar_factor)
        covar_kron_lv = KroneckerProductLazyTensor(task_var_lv, eye_lv)
        covariance_matrix = covar + covar_kron_lv
        covariance_matrix = add_diag(covariance_matrix, self.log_noise.exp())
        return input.__class__(mean, covariance_matrix)

    def variational_log_probability(self, input, target):

After Change

        noise = self.noise
        if covariance_matrix.ndimension() == 2:
            if settings.debug.on() and noise.size(0) > 1:
                raise RuntimeError("With batch_size > 1, expected a batched MultivariateNormal distribution.")
            noise = noise.squeeze(0)

        covariance_matrix = add_diag(covariance_matrix, noise)
        return input.__class__(mean, covariance_matrix)

Instances