Struct auto_diff::op::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 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 get_update_counter(&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.