numpy.diag_indices_from — NumPy v1.11 Manual (original) (raw)

numpy.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 : array, at least 2-D