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

use crate::dtype::RawDataType;
use crate::NdArray;

impl<T: RawDataType> PartialEq<NdArray<'_, T>> for NdArray<'_, T> {
    #[allow(clippy::op_ref)]
    fn eq(&self, other: &NdArray<T>) -> bool {
        &self == other
    }
}

impl<T: RawDataType> PartialEq<NdArray<'_, T>> for &NdArray<'_, T> {
    fn eq(&self, other: &NdArray<T>) -> bool {
        if self.shape != other.shape {
            return false;
        }
        self.flatiter().zip(other.flatiter()).all(|(a, b)| a == b)
    }
}

impl<T: RawDataType> PartialEq<&NdArray<'_, T>> for NdArray<'_, T> {
    fn eq(&self, other: &&NdArray<T>) -> bool {
        if self.shape != other.shape {
            return false;
        }
        self.flatiter().zip(other.flatiter()).all(|(a, b)| a == b)
    }
}