Function replace_scalar

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

Inplace replace in Array based on a condition

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

a = cond ? a : b; /// where cond, a are Arrays and b is scalar(f64)

§Parameters

  • a is the Array whose element will be replaced with element from b if corresponding element in cond Array is True
  • cond is the Array with conditional values
  • b is the scalar that will replace the element in output if corresponding element in cond Array is False

§Return Values

None