cupy.argwhere¶

cupy.argwhere(a)[source]

Return the indices of the elements that are non-zero.

Returns a (N, ndim) dimantional array containing the indices of the non-zero elements. Where N is number of non-zero elements and ndim is dimension of the given array.

Parameters

a (cupy.ndarray) – array

Returns

Indices of elements that are non-zero.

Return type

cupy.ndarray