DiagIndices =========== Return indices for accessing the main diagonal. **Abstract Signature:** ``DiagIndices(n: int, ndim: int = 2)`` .. raw:: html
jax.numpy.diag_indices
numpy.diag_indices