378c19e1422c00f91f7ce0e87de5b8f7d89799ed,gpytorch/lazy/kronecker_product_added_diag_lazy_tensor.py,KroneckerProductAddedDiagLazyTensor,inv_quad,#KroneckerProductAddedDiagLazyTensor#Any#Any#,55

Before Change

        res = inv_mat.matmul(res)
        res = q_matrix.matmul(res)

        return tensor.t().matmul(res).squeeze()

After Change

        res = q_matrix.transpose(-2, -1).matmul(tensor)
        res2 = inv_mat_sqrt.matmul(res)
        return res2.transpose(-2, -1).matmul(res2).sum((-1, -2))

Instances