jax.numpy.shape =============== .. currentmodule:: jax.numpy .. autofunction:: shape