acme-tensor 0.3.1

A comprehensive tensor library for Rust with support for automatic-differentiation.
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

/*
    Appellation: indexed <mod>
    Contrib: FL03 <jo3mccain@icloud.com>
*/
use crate::iter::LayoutIter;
use crate::tensor::TensorBase;

use super::Position;

pub struct IndexedIter<'a, T: 'a> {
    inner: LayoutIter,
    scope: Option<&'a T>,
    tensor: &'a TensorBase<T>,
}

impl<'a, T> IndexedIter<'a, T> {
    pub fn new(tensor: &'a TensorBase<T>) -> Self {
        Self {
            inner: tensor.layout().iter(),
            scope: None,
            tensor,
        }
    }
}

impl<'a, T> Iterator for IndexedIter<'a, T> {
    type Item = (&'a T, Position);

    fn next(&mut self) -> Option<Self::Item> {
        let pos = self.inner.next()?;
        self.scope = self.tensor.get_by_index(pos.index());
        self.scope.map(|scope| (scope, pos))
    }
}

impl<'a, T> DoubleEndedIterator for IndexedIter<'a, T> {
    fn next_back(&mut self) -> Option<Self::Item> {
        let pos = self.inner.next_back()?;
        self.scope = self.tensor.get_by_index(pos.index());
        self.scope.map(|scope| (scope, pos))
    }
}

impl<'a, T> ExactSizeIterator for IndexedIter<'a, T> {
    fn len(&self) -> usize {
        self.tensor.size()
    }
}