pub fn apply_unary_lanewise, V2: AsMut<[T2]> + Default>( x: V1, f: impl Fn(T1) -> T2, ) -> V2 { let mut y = V2::default(); assert_eq!(x.as_ref().len(), y.as_mut().len()); for (x, y) in x.as_ref().iter().zip(y.as_mut().iter_mut()) { *y = f(*x); } y } pub fn apply_binary_lanewise + AsMut<[T]> + Default>( a: V, b: V, f: impl Fn(T, T) -> T, ) -> V { let mut out = V::default(); let out_slice = out.as_mut(); let a_slice = a.as_ref(); let b_slice = b.as_ref(); for (o, (a, b)) in out_slice.iter_mut().zip(a_slice.iter().zip(b_slice.iter())) { *o = f(*a, *b); } out } pub fn apply_binary_scalar_rhs_lanewise + AsMut<[T]> + Default>( a: V, b: T, f: impl Fn(T, T) -> T, ) -> V { let mut out = V::default(); let out_slice = out.as_mut(); let a_slice = a.as_ref(); for (o, a) in out_slice.iter_mut().zip(a_slice.iter()) { *o = f(*a, b); } out } pub fn apply_binary_scalar_lhs_lanewise + AsMut<[T]> + Default>( a: T, b: V, f: impl Fn(T, T) -> T, ) -> V { let mut out = V::default(); let out_slice = out.as_mut(); let b_slice = b.as_ref(); for (o, b) in out_slice.iter_mut().zip(b_slice.iter()) { *o = f(a, *b); } out }