cunumeric.argwhere#

cunumeric.argwhere(a)#

Find the indices of array elements that are non-zero, grouped by element.

Parameters:

a (array_like) – Input data.

Returns:

index_array – Indices of elements that are non-zero. Indices are grouped by element. This array will have shape (N, a.ndim) where N is the number of non-zero items.

Return type:

ndarray

See also

numpy.argwhere

Availability:

Multiple GPUs, Multiple CPUs