Struct ComputationGraph 

pub struct ComputationGraph { /* private fields */ }

Computation graph for reverse-mode autodiff

Implementations

impl ComputationGraph

pub fn new() -> Self

Create a new computation graph

pub fn parameter(&mut self, name: String, value: f64) -> usize

Add a parameter node

pub fn constant(&mut self, value: Complex<f64>) -> usize

Add a constant node

pub fn add(&mut self, a: usize, b: usize) -> usize

Add two nodes

pub fn mul(&mut self, a: usize, b: usize) -> usize

Multiply two nodes

pub fn exp_i(&mut self, theta: usize) -> usize

Exponential of i times a real parameter

pub fn backward(&mut self, output: usize)

Backward pass to compute gradients

pub fn get_gradient(&self, param: &str) -> Option<f64>

Get gradient for a parameter

Trait Implementations

impl Debug for ComputationGraph

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.