348122d499073a7a58b21d6e5544bc4b47a9ee94,gpytorch/lazy/kronecker_product_added_diag_lazy_tensor.py,KroneckerProductAddedDiagLazyTensor,logdet,#KroneckerProductAddedDiagLazyTensor#,103

Before Change

        noise = self._diag_tensor[0, 0]
        sub_eigs = []
        for lazy_tensor in self._lazy_tensor.lazy_tensors:
            sub_eigs.append(lazy_tensor.evaluate().eig()[0][:, 0].unsqueeze(-1))

        eigs = sub_eigs[0].matmul(sub_eigs[1].t())
        return torch.log(eigs + noise).sum()

After Change

        return inv_quad_term, logdet_term

    def logdet(self):
        noisy_eigenvalues = self._kron_eigenvalues.diag() + self._diag_tensor.diag()
        return torch.log(noisy_eigenvalues).sum()

    @property
    @cached

Instances