dpnp.triu_indices

dpnp.triu_indices(n, k=0, m=None)

Return the indices for the upper-triangle of an (n, m) array.

Parameters:

• n (int) -- The size of the arrays for which the returned indices will be valid.

• k (int, optional) -- Diagonal offset (see triu for details).

• m (int, optional) -- The column dimension of the arrays for which the returned arrays will be valid. By default m is taken equal to n.

Returns:

inds -- The indices for the triangle. The returned tuple contains two arrays, each with the indices along one dimension of the array. Can be used to slice a ndarray of shape(n, n).

Return type:

tuple, shape(2) of ndarrays, shape(n)