Struct Op

pub struct Op { /* private fields */ }

Op is the Rc wrapper of typed op trait

Implementations

impl Op

pub fn new(op: Rc<RefCell<Box<dyn OpTrait>>>) -> Self

pub fn inner(&self) -> &Rc<RefCell<Box<dyn OpTrait>>>

pub fn ref_copy(&self) -> Self

pub fn get_name(&self) -> String

pub fn get_input_size(&self) -> usize

pub fn get_output_size(&self) -> usize

pub fn apply(&self, input: &[Tensor], output: &[Tensor])

Read the input, do the calculation and write result to output. Called by compute_grapyh.

pub fn grad( &self, input: &[Tensor], output_grad: &[Tensor], input_grad: &[Tensor], )

Given input and output_grad, return input_grad (forward view) Called by compute_grapyh.

pub fn get_values(&self) -> Vec<Tensor>

access weight/paramenters

pub fn set_values(&self, v: &[Tensor])

set parameters

pub fn get_grads(&self) -> Vec<Tensor>

return gradient for weight/parameters.

Trait Implementations

impl Clone for Op

fn clone(&self) -> Op

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'de> Deserialize<'de> for Op

fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for Op

fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>

where S: Serializer,

Serialize this value into the given Serde serializer.