Struct ComputationGraph 

pub struct ComputationGraph<T: TensorElement> { /* private fields */ }

Computation graph

Implementations

impl<T: TensorElement + Copy> ComputationGraph<T>

pub fn new() -> Self

Create a new empty computation graph

pub fn constant(&mut self, tensor: Tensor<T>) -> NodeId

Add a constant tensor to the graph

pub fn binary_op( &mut self, op: GraphOp, left: NodeId, right: NodeId, device: DeviceType, ) -> NodeId

Add a binary operation node

pub fn unary_op( &mut self, op: GraphOp, input: NodeId, device: DeviceType, ) -> NodeId

Add a unary operation node

pub fn mark_output(&mut self, node: NodeId)

Mark a node as an output

pub fn num_nodes(&self) -> usize

Get the number of nodes

pub fn num_outputs(&self) -> usize

Get the number of output nodes

pub fn topological_sort(&self) -> Result<Vec<NodeId>>

Perform topological sort of the graph

pub fn optimize(&mut self) -> Result<()>

where T: Add<Output = T> + Sub<Output = T> + Mul<Output = T> + Div<Output = T> + FloatElement,

Optimize the graph by fusing operations

pub fn execute(&self) -> Result<Vec<Tensor<T>>>

where T: Add<Output = T> + Sub<Output = T> + Mul<Output = T> + Div<Output = T> + FloatElement,

Execute the graph and get outputs

pub fn to_dot(&self) -> String

Generate DOT representation for visualization

Trait Implementations

impl<T: TensorElement + Copy> Default for ComputationGraph<T>

fn default() -> Self

Returns the “default value” for a type.