DiagIndicesFrom

Return indices for accessing the main diagonal of a given array.

Abstract Signature:

DiagIndicesFrom(arr: Tensor)

JAX (Core)

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

NumPy

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