d37088fd9a08cc0972a64f5884b0647540309cb9,gpytorch/lazy/inv_quad_lazy_variable.py,InvQuadLazyVariable,diag,#InvQuadLazyVariable#,69

Before Change

    def diag(self):
        lhs = self.left_mat
        rhs = inv_matmul(self.inv_mat, self.right_mat.transpose(-1, -2))
        return (lhs * rhs.transpose(-1, -2)).sum(-1)

After Change

    def diag(self):
        lhs = self.left_mat
        rhs = inv_matmul(self.inv_mat, self.right_mat.transpose(-1, -2))
        res = (lhs * rhs.transpose(-1, -2)).sum(-1)
        res.add_(self.added_diag)
        return res

    def inv_quad_log_det(self, inv_quad_rhs=None, log_det=False):
        if inv_quad_rhs is not None and inv_quad_rhs.ndimension() == 1:
            inv_quad_rhs = inv_quad_rhs.unsqueeze(-1)

Instances