select

arrayfire

Function select 

Source 
pub fn select<T>(a: &Array<T>, cond: &Array<bool>, b: &Array<T>) -> Array<T>
where
    T: HasAfEnum,
Expand description

Element wise conditional operator for Arrays

This function does the C-equivalent of the following statement, except that the operation happens on a GPU for all elements simultaneously.

c = cond ? a : b; /// where cond, a & b are all objects of type Array

§Parameters

  • a is the Array whose element will be assigned to output if corresponding element in cond Array is True
  • cond is the Array with boolean values
  • b is the Array whose element will be assigned to output if corresponding element in cond Array is False

§Return Values

An Array