cupy.where

cupy.where(condition, x=None, y=None)[source]

Return elements, either from x or y, depending on condition.

If only condition is given, return condition.nonzero().

Parameters:
  • condition (cupy.ndarray) – When True, take x, otherwise take y.
  • x (cupy.ndarray) – Values from which to choose on True.
  • y (cupy.ndarray) – Values from which to choose on False.
Returns:

Each element of output contains elements of x when

condition is True, otherwise elements of y. If only condition is given, return the tuple condition.nonzero(), the indices where condition is True.
Return type:

cupy.ndarray

Warning

This function may synchronize the device if both x and y are omitted.

See also

numpy.where()