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

use crate::buffer_iterator::BufferIterator;
use crate::dtype::RawDataType;
use crate::NdArray;

pub struct FlatIterator<T: RawDataType> {
    buffer_iterator: BufferIterator<T>,
}

impl<T: RawDataType> FlatIterator<T> {
    pub(super) fn from(tensor: &NdArray<T>) -> Self {
        Self {
            buffer_iterator: BufferIterator::from(tensor),
        }
    }
}

impl<T: RawDataType> Iterator for FlatIterator<T> {
    type Item = T;

    fn next(&mut self) -> Option<Self::Item> {
        self.buffer_iterator.next().map(|ptr| unsafe { *ptr })
    }
}