Method unique

ot/backend.py:1707–1708

(self, a, return_inverse=False)

        return jnp.diag(a, k)

    def unique(self, a, return_inverse=False):
        return jnp.unique(a, return_inverse=return_inverse)

    def stack(self, arrays, axis=0):
        return jnp.stack(arrays, axis)