Skip to main content

where_

nuvec_core::ops

Function where_ 

Source 
pub fn where_(
    condition: impl Into<ScalarOrArray>,
    a: impl Into<ScalarOrArray>,
    b: impl Into<ScalarOrArray>,
) -> Array
Expand description

Element-wise selection: a where condition is true, else b (MLX’s where); see Array::select. Any operand may be a weak Rust number.