Method stack

ot/backend.py:1710–1711

(self, arrays, axis=0)

        return jnp.unique(a, return_inverse=return_inverse)

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

    def reshape(self, a, shape):
        return jnp.reshape(a, shape)