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
32
33
34
35

// Trait to check if a vector has homogeneously sized (rectangular) dimensions
// [[0], [1], [2]] - homogenous
// [[0, 1], [2]] - not homogenous

use crate::recursive_trait_base_cases;
use crate::util::shape::Shape;

pub(crate) trait Homogenous {
    fn check_homogenous(&self) -> bool;
}

impl<T: Homogenous + Shape> Homogenous for Vec<T> {
    fn check_homogenous(&self) -> bool {
        let first_shape = self[0].shape();
        self.iter().all(|v| v.shape() == first_shape)
    }
}

impl<T, const N: usize> Homogenous for [T; N] {
    fn check_homogenous(&self) -> bool {
        true
    }
}

macro_rules! homogenous_vec_trait {
    ( $dtype: ty ) => {
        impl Homogenous for Vec<$dtype> {
            fn check_homogenous(&self) -> bool {
                true
            }
        }
    };
}

recursive_trait_base_cases!(homogenous_vec_trait);