Method eigh

ot/backend.py:1848–1849

(self, a)

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

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

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