Method eigh

ot/backend.py:2405–2406

(self, a)

        return torch.einsum("...jk,...kl->...jl", Q, torch.transpose(V, -1, -2))

    def eigh(self, a):
        return torch.linalg.eigh(a)

    def kl_div(self, p, q, mass=False, eps=1e-16):
        value = torch.sum(p * torch.log(p / q + eps))