Method meshgrid

ot/backend.py:1701–1702

(self, a, b)

            )

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

    def diag(self, a, k=0):
        return jnp.diag(a, k)