Struct auto_diff::compute_graph::Net

pub struct Net { /* private fields */ }

The computation network. Connection has duplication.

Implementations

impl Net

pub fn new() -> Net

pub fn get_data(&self) -> &GenIndex<Tensor>

pub fn get_data_mut(&mut self) -> &mut GenIndex<Tensor>

pub fn get_ops(&self) -> &GenIndex<Op>

pub fn get_ops_mut(&mut self) -> &mut GenIndex<Op>

pub fn add_tensor(&mut self, t: Tensor) -> GenKey

pub fn get_tensor(&self, id: GenKey) -> Result<Tensor, AutoDiffError>

pub fn set_tensor(
    &mut self,
    id: GenKey,
    val: Tensor
) -> Result<(), AutoDiffError>

pub fn add_op(&mut self, op: Op) -> GenKey

Insert operator into the network.

pub fn get_op(&self, id: GenKey) -> Result<Op, AutoDiffError>

pub fn get_grad(&self, id: GenKey) -> Result<Tensor, AutoDiffError>

pub fn get_input_edge_data(&self) -> BTreeSet<GenKey>

pub fn get_output_edge_data(&self) -> BTreeSet<GenKey>

pub fn connect(&mut self, input: &[GenKey], op: GenKey, output: &[GenKey])

Build input-operator-output relation, with given components.

pub fn set_mark(&mut self, did: &GenKey)

set the set_mark, set_mark is used to label var with input value with it.

pub fn unset_mark(&mut self, did: &GenKey)

pub fn eval(&mut self, starting_node: &[GenKey]) -> Result<(), BTreeSet<GenKey>>

Forward evaluate the computaiton graph.

pub fn bptt(&mut self, output_grad: &BTreeMap<GenKey, Tensor>)

pub fn visit_op<F>(
    &mut self,
    closure: F,
    allow: Option<Vec<GenKey>>,
    skip: Option<Vec<GenKey>>
) where
    F: Fn(&Op), 

Iterate over all ops, no order guarantee

pub fn visit_data<F>(&mut self, closure: F) where
    F: Fn(GenKey, &Tensor), 

pub fn append(
    &mut self,
    other: &Self,
    original_keys: &[GenKey]
) -> Result<Vec<GenKey>, AutoDiffError>

Move content in other network into self. Return new ids for those have origianl_keys in the old network.

Trait Implementations

impl Clone for Net

fn clone(&self) -> Net

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Net

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

Formats the value using the given formatter.

impl Default for Net

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for Net

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where
    __D: Deserializer<'de>, 

Deserialize this value from the given Serde deserializer.

impl Serialize for Net

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error> where
    __S: Serializer, 

Serialize this value into the given Serde serializer.