deep_core 0.1.1

deep prototype
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

use std::ops::{Index, IndexMut};
use std::vec::*;

use math::{Shape};

pub enum Error {
    ShapeMismatch
}

pub fn position_to_index(position: &[usize], shape: &Shape) -> usize {
    let mut result = position[0];
    let mut shape_multiplier = 1;
    for component_index in 0..position.len() - 1 {
        shape_multiplier *= shape[component_index];
        result += position[component_index + 1] * shape_multiplier;
    }

    result
}

// -- Tensor

#[derive(Debug)]
pub struct Tensor<T> {
    data: Vec<T>,
    shape: Shape,
}

impl<T> Tensor<T> {
    pub fn get_shape(&self) -> &Shape {
        &self.shape
    }
}

impl<T: Default> Tensor<T> {
    pub fn default<IntoShape: Into<Shape>>(into_shape: IntoShape) -> Self {
        let shape = into_shape.into();
        let size = shape.size();

        let mut data = Vec::with_capacity(size);
        for _ in 0..size {
            data.push(T::default());
        }

        Tensor {
            shape, data
        }
    }
}

// Iterator

#[derive(Debug)]
pub struct TensorIterationItem<'a, T: 'a> {
    pub value: &'a T,
    pub index: usize,
}

pub struct TensorIterator<'a, T: 'a> {
    pub tensor: &'a Tensor<T>,
    pub index: usize,
}

impl<'a, T: 'a> Iterator for TensorIterator<'a, T> {
    type Item = TensorIterationItem<'a, T>;
    fn next(&mut self) -> Option<Self::Item> {
        if self.index < self.tensor.shape.size() {
            let value = &self.tensor.data[self.index];
            self.index += 1;
            Some(TensorIterationItem{
                index: self.index,
                value, 
            })
        } else {
            None
        }
    }
}

impl<'a, T: 'a> IntoIterator for &'a Tensor<T> {
    type Item = TensorIterationItem<'a, T>;
    type IntoIter = TensorIterator<'a, T>;
    fn into_iter(self) -> Self::IntoIter {
        TensorIterator { index: 0, tensor: self }
    }
}

// -- Index

impl<T> Index<usize> for Tensor<T> {
    type Output = T;
    fn index(&self, index: usize) -> &Self::Output {
        &self.data[index]
    }
}

impl<T> IndexMut<usize> for Tensor<T> {
    fn index_mut<'a>(&'a mut self, index: usize) -> &'a mut T {
        &mut self.data[index]
    }
}

impl<T> Index<Vec<usize>> for Tensor<T> {
    type Output = T;
    fn index(&self, position: Vec<usize>) -> &Self::Output {
        let index = position_to_index(&position, &self.shape);
        &self.data[index]
    }
}

impl<T> IndexMut<Vec<usize>> for Tensor<T> {
    fn index_mut<'a>(&'a mut self, position: Vec<usize>) -> &'a mut T {
        let index = position_to_index(&position, &self.shape);
        &mut self.data[index]
    }
}