Struct AutoDiff 

pub struct AutoDiff { /* private fields */ }

Automatic differentiation engine

Implementations

impl AutoDiff

pub fn new() -> Self

Create a new AutoDiff engine

pub fn register_parameter(&mut self, param: DifferentiableParam)

Register a parameter

pub fn set_graph(&mut self, graph: ComputationNode)

Set computation graph

pub fn forward(&mut self) -> Result<f64>

Forward pass - compute value

pub fn backward(&mut self, loss_gradient: f64) -> Result<()>

Backward pass - compute gradients

pub fn gradients(&self) -> HashMap<String, f64>

Get all gradients

pub fn update_parameters(&mut self, learning_rate: f64)

Update parameters using gradients