jax.numpy.diag ============== .. currentmodule:: jax.numpy .. autofunction:: diag