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:
See also