jax.numpy.diagonal ================== .. currentmodule:: jax.numpy .. autofunction:: diagonal