jax.numpy.expand\_dims ====================== .. currentmodule:: jax.numpy .. autofunction:: expand_dims