ferrite-dl 0.2.0

Deep learning library written in pure 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
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

use crate::*;  // Import from parent module's base.rs

use std::fmt;

pub trait Display {
  fn print(&self);
  fn print_data_recursive<'a>(data: &'a [f32], shape: &'a [usize], stride: &'a [usize]) -> String;
  fn print_data(&self);
}

impl fmt::Display for CpuStorage {
  fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
    write!(f, "{}", Self::print_data_recursive(&self.data().read().unwrap(), self.shape(), self.stride()))
  }
}

impl fmt::Debug for CpuStorage {
  fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
    write!(f, "{}", Self::print_data_recursive(&self.data().read().unwrap(), self.shape(), self.stride()))
  }
}

impl Display for CpuStorage {
  fn print(&self) {
    println!("Data: {:?}", self.data());
    println!("Shape: {:?}", self.shape());
    println!("Strides: {:?}", self.stride());
  }

  fn print_data_recursive<'a>(data: &'a [f32], shape: &'a [usize], stride: &'a [usize]) -> String {
    let mut res = String::new();
    res += "[";
    if shape.len() == 1 {
      for i in 0..shape[0] {
        res += &format!("{}", data[i*stride[0]]);

        if i < shape[0] - 1 {
          res += ", ";
        }
        
      }
    } else {
      for i in 0..shape[0] {
        let start = i*stride[0];
        let sub_res = Self::print_data_recursive(&data[start..], &shape[1..], &stride[1..]);
        res += &sub_res;
        if i < shape[0] - 1 {
          res += ", ";
        }
      }
    }

    res += "]";
    res
  }

  fn print_data(&self) {
    let res = Self::print_data_recursive(&self.data().read().unwrap(), self.shape(), self.stride());
    println!("{}", res);
  }
}