cunumeric.diag_indices_from

cunumeric.diag_indices_from(arr: ndarray) → tuple[ndarray, ...]

Return the indices to access the main diagonal of an n-dimensional array.

See diag_indices for full details.

Parameters:

arr (array_like) – at least 2-D

See also

numpy.diag_indices_from, numpy.diag_indices

Availability:

Multiple GPUs, Multiple CPUs