Struct DynamicGraph 

pub struct DynamicGraph { /* private fields */ }

Dynamic computation graph

Implementations

impl DynamicGraph

pub fn new() -> Self

Create a new dynamic graph

pub fn start_recording(&self)

Start recording operations

pub fn stop_recording(&self)

Stop recording operations

pub fn is_recording(&self) -> bool

Check if recording

pub fn add_node(&self, op: String, inputs: Vec<usize>, output: Tensor) -> usize

Add a node to the graph

pub fn get_node(&self, id: usize) -> Option<GraphNode>

Get a node by ID

pub fn clear(&self)

Clear the graph

pub fn num_nodes(&self) -> usize

Get number of nodes

pub fn backward(&self, node_id: usize, grad: Tensor) -> HashMap<usize, Tensor>

Perform backward pass from a node

Trait Implementations

impl Debug for DynamicGraph

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

Formats the value using the given formatter.

impl Default for DynamicGraph

fn default() -> Self

Returns the “default value” for a type.