Method sort2

ot/backend.py:2432–2433

(self, a, axis=-1)

        return torch.prod(a, dim=axis)

    def sort2(self, a, axis=-1):
        return torch.sort(a, axis)

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