gtensor 1.0.0

Reverse-mode autodifferentiation of computational graphs with tensors and more for machine learning.
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


use std::fmt::Display;
use std::fmt::Formatter;

use serde::{Serialize, Deserialize};
use anyhow::Result;

use crate::operators::Operator;
use crate::tensor::shape::Shape;
use crate::graph::node::Node;

pub trait Optimizer: Operator {
    fn to_operator(&self, shape: Shape) -> Box<dyn Operator>;
}

mod sgd;
mod adam;
mod momentum;

pub mod opt {
    use super::*;

    pub use sgd::sgd;
    pub use adam::adam;
    pub use momentum::momentum;
}