DiagIndices

Return indices for accessing the main diagonal.

Abstract Signature:

DiagIndices(n: int, ndim: int = 2)

JAX (Core)

API: jax.numpy.diag_indices
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.diag_indices
Strategy: Direct Mapping
Official Docs ↗