Method kl_div

ot/backend.py:2408–2412

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

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

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