cupy.extract#

cupy.extract(condition, a)[source]#

Return the elements of an array that satisfy some condition.

This is equivalent to np.compress(ravel(condition), ravel(arr)). If condition is boolean, np.extract is equivalent to arr[condition].

Parameters:

  • condition (int or array_like) – An array whose nonzero or True entries indicate the elements of array to extract.

  • a (cupy.ndarray) – Input array of the same size as condition.

Returns:

Rank 1 array of values from arr where condition is True.

Return type:

cupy.ndarray

Warning

This function may synchronize the device.

See also

numpy.extract()