dendritic_autodiff::node

Trait Node

Source 
pub trait Node {
    // Required methods
    fn forward(&mut self);
    fn backward(&mut self, upstream_gradient: NDArray<f64>);
    fn value(&self) -> NDArray<f64>;
    fn grad(&self) -> NDArray<f64>;
    fn set_grad(&mut self, upstream_gradient: NDArray<f64>);
}
Expand description

Methods for each value in computation graph

Required Methods§

Source

fn forward(&mut self)

Source

fn backward(&mut self, upstream_gradient: NDArray<f64>)

Source

fn value(&self) -> NDArray<f64>

Source

fn grad(&self) -> NDArray<f64>

Source

fn set_grad(&mut self, upstream_gradient: NDArray<f64>)

Implementors§

Source§

impl Node for Value<NDArray<f64>>

Source§

impl<LHS, RHS> Node for Regularization<RHS, LHS>
where RHS: Node, LHS: Node,

Source§

impl<LHS, RHS> Node for ScaleAdd<RHS, LHS>
where RHS: Node, LHS: Node,

Source§

impl<LHS, RHS> Node for L1Regularization<RHS, LHS>
where RHS: Node, LHS: Node,

Source§

impl<LHS, RHS> Node for L2Regularization<RHS, LHS>
where RHS: Node, LHS: Node,

Source§

impl<RHS, LHS> Node for Dot<RHS, LHS>
where RHS: Node, LHS: Node,