chela 0.0.2

High-performance Machine Learning, Auto-Differentiation and Tensor Algebra crate for Rust
Documentation 
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

use crate::traits::binary_ops::BinaryOps;
use crate::{NdArray, RawDataType};
use std::ops::{Add, BitAnd, BitOr, Div, Mul, Rem, Shl, Shr, Sub};

use paste::paste;

macro_rules! implement_binary_ops {
    ($object:ident, $($trait_: ident, $operator:tt, $method: ident;)* ) => { $(
        impl<T: RawDataType + $trait_<Output=T>> $trait_<$object<'_, T>> for $object<'_, T> {
            type Output = $object<'static, T>;

            fn $method(self, rhs: $object<T>) -> Self::Output { &self $operator &rhs }
        }

        impl<T: RawDataType + $trait_<Output=T>> $trait_<&$object<'_, T>> for $object<'_, T> {
            type Output = $object<'static, T>;

            fn $method(self, rhs: &$object<T>) -> Self::Output { &self $operator rhs }
        }
        
        impl<T: RawDataType + $trait_<Output=T>> $trait_<$object<'_, T>> for &$object<'_, T> {
            type Output = $object<'static, T>;

            fn $method(self, rhs: $object<T>) -> Self::Output { self $operator &rhs }
        }

        impl<T: RawDataType + $trait_<Output=T>> $trait_<&$object<'_, T>> for &$object<'_, T> {
            type Output = $object<'static, T>;

            fn $method(self, rhs: &$object<T>) -> Self::Output { <T as BinaryOps<T>>::$method(self, rhs) }
        }
        
        impl<T: RawDataType + $trait_<Output=T>> $trait_<T> for $object<'_, T> {
            type Output = $object<'static, T>;

            fn $method(self, rhs: T) -> Self::Output { paste! { &self $operator rhs } }
        }

        impl<T: RawDataType + $trait_<Output=T>> $trait_<T> for &$object<'_, T> {
            type Output = $object<'static, T>;

            fn $method(self, rhs: T) -> Self::Output { paste! { <T as BinaryOps<T>>::[<$method _scalar>](self, rhs) } }
        }
    )*};

    ($dtype1:ty, $dtype2:ty, $($trait_: ident, $method: ident;)* ) => {
        implement_binary_ops!($dtype1, $($trait_, $method;)* );
        implement_binary_ops!($dtype2, $($trait_, $method;)* );
    };

    ($dtype1:ty, $dtype2:ty, $dtype3:ty, $dtype4:ty, $($trait_: ident, $method: ident;)* ) => {
        implement_binary_ops!($dtype1, $dtype2, $($trait_, $method;)* );
        implement_binary_ops!($dtype3, $dtype4, $($trait_, $method;)* );
        implement_binary_ops!($dtype5, $dtype6, $($trait_, $method;)* );
    }
}

implement_binary_ops!(
    NdArray,
    Add, +, add;
    Sub, -, sub;
    Mul, *, mul;
    Div, /, div;
    Rem, %, rem;
    BitAnd, &, bitand;
    BitOr, |, bitor;
    Shl, <<, shl;
    Shr, >>, shr;
);