Skip to main content

zip_map3_into

strided_kernel

Function zip_map3_into 

Source 
pub fn zip_map3_into<D, A, B, C, OpA, OpB, OpC>(
    dest: &mut StridedViewMut<'_, D>,
    a: &StridedView<'_, A, OpA>,
    b: &StridedView<'_, B, OpB>,
    c: &StridedView<'_, C, OpC>,
    f: impl Fn(A, B, C) -> D + Sync,
    exec: Exec<'_>,
) -> Result<(), StridedError>
where
    D: Copy + Send + Sync,
    A: Copy + Send + Sync,
    B: Copy + Send + Sync,
    C: Copy + Send + Sync,
    OpA: ElementOp<A>,
    OpB: ElementOp<B>,
    OpC: ElementOp<C>,
Expand description

Ternary element-wise operation: dest[i] = f(a[i], b[i], c[i]).