Trait auto_diff::op::Op

pub trait Op {
    fn get_name(&self) -> &str;
    fn apply(
        &mut self, 
        input: &Vec<Rc<RefCell<Tensor>>>, 
        output: &mut Vec<Rc<RefCell<Tensor>>>
    );
    fn grad(&self, input: u32, output: u32);
}

Required methods

fn get_name(&self) -> &str

fn apply(
    &mut self,
    input: &Vec<Rc<RefCell<Tensor>>>,
    output: &mut Vec<Rc<RefCell<Tensor>>>
)

fn grad(&self, input: u32, output: u32)

Loading content...

Implementors

impl Op for Linear

fn get_name(&self) -> &str

fn apply(
    &mut self,
    input: &Vec<Rc<RefCell<Tensor>>>,
    output: &mut Vec<Rc<RefCell<Tensor>>>
)

fn grad(&self, input: u32, output: u32)

impl Op for add

fn get_name(&self) -> &str

fn apply(
    &mut self,
    input: &Vec<Rc<RefCell<Tensor>>>,
    output: &mut Vec<Rc<RefCell<Tensor>>>
)

fn grad(&self, input: u32, output: u32)

impl Op for div

fn get_name(&self) -> &str

fn apply(
    &mut self,
    input: &Vec<Rc<RefCell<Tensor>>>,
    output: &mut Vec<Rc<RefCell<Tensor>>>
)

fn grad(&self, input: u32, output: u32)

impl Op for mul

fn get_name(&self) -> &str

fn apply(
    &mut self,
    input: &Vec<Rc<RefCell<Tensor>>>,
    output: &mut Vec<Rc<RefCell<Tensor>>>
)

fn grad(&self, input: u32, output: u32)

impl Op for sub

fn get_name(&self) -> &str

fn apply(
    &mut self,
    input: &Vec<Rc<RefCell<Tensor>>>,
    output: &mut Vec<Rc<RefCell<Tensor>>>
)

fn grad(&self, input: u32, output: u32)

Loading content...