cupynumeric.nonzero

cupynumeric.nonzero(

a: ndarray,

) → tuple[ndarray, ...]

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 (array_like) – Input array.

Returns:

tuple_of_arrays – Indices of elements that are non-zero.

Return type:

tuple

See also

numpy.nonzero

Availability:

Multiple GPUs, Multiple CPUs