dorset 0.1.0

A rust implementation of Stan Math automatic differentiation library.
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

//! Off-the-shelve memory arena using `typed-arena`
//! crate.
use typed_arena;
pub use core::vari::*;
pub use std::rc::Rc;

pub struct ChainStack {
    pub arena: typed_arena::Arena<Vari>,
    pub stack: Vec<Rc<*mut Vari>>,
}
impl ChainStack {
    pub fn new() -> ChainStack {
        ChainStack {
            arena: typed_arena::Arena::new(),
            stack: vec![],
        }
    }
    pub fn alloc(&mut self, vi: Vari) -> Rc<*mut Vari> {
        let v: Rc<*mut Vari> = Rc::new(self.arena.alloc(vi));
        self.stack.push(v.clone());
        v
    }
    pub fn chain(&self) {
        for it in self.stack.iter().rev() {
            let vi: &mut Vari = it.clone().into();
            vi.chain();
        }
    }
    pub fn set_zero_all_adjoints(&self) {
        for it in self.stack.iter() {
            let vi: &mut Vari = it.clone().into();
            vi.set_zero_adjoint();
        }
    }
}