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