redstone-ml 0.0.0

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

use crate::{NdArray, Tensor, TensorDataType};

impl<T: TensorDataType> PartialEq<Tensor<'_, T>> for Tensor<'_, T> {
    fn eq(&self, other: &Tensor<T>) -> bool {
        *self.array == *other.array
    }
}

impl<T: TensorDataType> PartialEq<NdArray<'_, T>> for Tensor<'_, T> {
    fn eq(&self, other: &NdArray<T>) -> bool {
        *self.array == other
    }
}

impl<T: TensorDataType> PartialEq<Tensor<'_, T>> for NdArray<'_, T> {
    fn eq(&self, other: &Tensor<T>) -> bool { 
        self == *other.array
    }
}

impl<T: TensorDataType> PartialEq<&Tensor<'_, T>> for NdArray<'_, T> {
    fn eq(&self, other: &&Tensor<T>) -> bool {
        self == *other.array
    }
}

impl<T: TensorDataType> PartialEq<Tensor<'_, T>> for &NdArray<'_, T> {
    fn eq(&self, other: &Tensor<T>) -> bool {
        *self == *other.array
    }
}