cupy.nonzero¶
- cupy.nonzero(a)[source]¶
Return the indices of the elements that are non-zero.
Returns a tuple of arrays, one for each dimension of a, containing the indices of the non-zero elements in that dimension.
- Parameters
a (cupy.ndarray) – array
- Returns
Indices of elements that are non-zero.
- Return type
tuple of arrays
Warning
This function may synchronize the device.
See also