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

See also

numpy.argwhere()