cupy.diag_indices_from#

cupy.diag_indices_from(arr)[source]#

Return the indices to access the main diagonal of an n-dimensional array. See diag_indices for full details.

Parameters:

arr (cupy.ndarray) – At least 2-D.

See also

numpy.diag_indices_from()