Struct primitiv::Graph

pub struct Graph { /* fields omitted */ }

Computation graph.

Methods

impl Graph

pub fn new() -> Self

Creates a new Graph object.

pub fn set_default(graph: &mut Self)

Specifies a new default graph.

pub fn clear(&mut self)

Clear all operators in the graph.

Remark: After calling this method, all Node objects supplied by the graph itself is invalidated.

pub fn forward(&mut self, node: &Node) -> Tensor

Calculates the value of given node.

pub fn backward(&mut self, node: &Node)

Calculates the backpropagation.

pub fn get_shape(&self, node: &Node) -> Shape

Retrieves the shape of the node.

pub fn get_device(&self, node: &Node) -> AnyDevice

Retrieves the device of the node.

pub fn dump(&self, format: &str) -> String

Dump internal graph structure.

Available options:

• “dot” … Graphviz’s dot format.

pub fn num_operators(&self) -> u32

Returns the number of operators in the computation graph.

Trait Implementations

impl Debug for Graph

fn fmt(&self, __arg_0: &mut Formatter) -> Result

Formats the value using the given formatter.

impl Wrap<primitivGraph_t> for Graph

fn from_raw(ptr: *mut primitivGraph_t, owned: bool) -> Self

fn as_ptr(&self) -> *const primitivGraph_t

fn as_mut_ptr(&mut self) -> *mut primitivGraph_t

fn is_owned(&self) -> bool

impl Drop for Graph

fn drop(&mut self)

Executes the destructor for this type.