Method kl_div

ot/backend.py:1851–1855

(self, p, q, mass=False, eps=1e-16)

        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))
        if mass:
            value = value + jnp.sum(q - p)
        return value

    def isfinite(self, a):
        return jnp.isfinite(a)