cunumeric.nonzero#

cunumeric.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