1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

extern crate alloc;
use alloc::vec::Vec;

pub fn element_wise_op<T: Copy + Default, F>(lhs: &[T], rhs: &[T], f: F) -> Vec<T>
where
    F: Fn(T, T) -> T,
{
    let len = core::cmp::min(lhs.len(), rhs.len());
    let lhs = &lhs[..len];
    let rhs = &rhs[..len];

    let mut vec = vec![T::default(); len];

    let out_slice = &mut vec[..len];

    for idx in 0..len {
        out_slice[idx] = f(lhs[idx], rhs[idx])
    }
    vec
}

pub fn element_wise_op_mut<T: Copy, F>(lhs: &[T], rhs: &[T], out: &mut [T], f: F)
where
    F: Fn(T, T) -> T,
{
    let len = core::cmp::min(lhs.len(), rhs.len());
    let lhs = &lhs[..len];
    let rhs = &rhs[..len];

    let out_slice = &mut out[..len];

    for idx in 0..len {
        out_slice[idx] = f(lhs[idx], rhs[idx])
    }
}