cupy.select#

cupy.select(condlist, choicelist, default=0)[source]#

Return an array drawn from elements in choicelist, depending on conditions.

Parameters:
  • condlist (list of bool arrays) – The list of conditions which determine from which array in choicelist the output elements are taken. When multiple conditions are satisfied, the first one encountered in condlist is used.

  • choicelist (list of cupy.ndarray) – The list of arrays from which the output elements are taken. It has to be of the same length as condlist.

  • default (scalar) – If provided, will fill element inserted in output when all conditions evaluate to False. default value is 0.

Returns:

The output at position m is the m-th element of the array in choicelist where the m-th element of the corresponding array in condlist is True.

Return type:

cupy.ndarray

See also

numpy.select()