Struct 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.

pub fn set_label( &mut self, label: &str, id: &GenKey, ) -> Result<(), AutoDiffError>

For introspection.

pub fn get_id_by_label(&self, label: &str) -> Result<GenKey, AutoDiffError>

pub fn drop_label(&mut self, label: &str) -> Result<GenKey, AutoDiffError>

Trait Implementations

impl Clone for Net

fn clone(&self) -> Net

Returns a duplicate 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.