Method flip

ot/backend.py:1657–1658

(self, a, axis=None)

            return jax.vmap(lambda b, u: jnp.searchsorted(b, u, side))(a, v)

    def flip(self, a, axis=None):
        return jnp.flip(a, axis)

    def outer(self, a, b):
        return jnp.outer(a, b)