Struct WhereArgs 

pub struct WhereArgs<'a, T: Element, const N: usize> {
    pub cond: TensorRef<'a, u8, N>,
    pub a: TensorRef<'a, T, N>,
    pub b: TensorRef<'a, T, N>,
    pub y: TensorMut<'a, T, N>,
}

Args bundle for a where launch.

cond is u8 (0 = false, non-zero = true). a, b, y share dtype T. All four operands can broadcast independently to y.shape via stride-0 axes.

Fields

cond: TensorRef<'a, u8, N>

Boolean mask. 0u8 selects b, any other value selects a.

a: TensorRef<'a, T, N>

Value selected where cond != 0.

b: TensorRef<'a, T, N>

Value selected where cond == 0.

y: TensorMut<'a, T, N>

Output.