dsalgo 0.3.10

A package for Datastructures and Algorithms.
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

pub trait Shape {
    fn shape() -> &'static [usize];
}

pub trait Size {
    fn size() -> usize;
}

impl<T: Shape> Size for T {
    fn size() -> usize {
        let mut size = 1;

        for &dim in Self::shape() {
            size *= dim;
        }

        size
    }
}

pub trait Dimension {
    fn dimension() -> usize;
}

impl<P: Shape> Dimension for P {
    fn dimension() -> usize {
        P::shape().len()
    }
}

pub trait Strides {
    fn strides() -> Vec<usize>;
}

impl<P: Shape> Strides for P {
    fn strides() -> Vec<usize> {
        let ndim = Self::dimension();

        let mut strides: Vec<usize> = P::shape().into();

        if ndim > 0 {
            strides[ndim - 1] = 1;
        }

        for i in (1..ndim).rev() {
            strides[i - 1] = strides[i] * P::shape()[i];
        }

        strides
    }
}

pub trait TensorProperty: Shape {}

impl<P: Shape> TensorProperty for P {}