Struct auto_diff::op::linear::Linear

[+] Show declaration

pub struct Linear { /* fields omitted */ }

Implementations

impl Linear[−]

pub fn new(
    in_features: Option<usize>,
    out_features: Option<usize>,
    bias: bool
) -> Linear

pub fn weight(&self) -> &Tensor

pub fn bias(&self) -> &Tensor

Trait Implementations

impl OpTrait for Linear[+]

[+] Show hidden undocumented items

fn get_name(&self) -> String

fn get_input_size(&self) -> usize

fn get_output_size(&self) -> usize

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

Forward pass

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

Given the forward input value and backward output_grad, Update weight gradient. return backward input gradeint.

fn get_values(&self) -> Vec<&Tensor>[−]

access weight values

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

fn get_grads(&self) -> Vec<&Tensor>[−]

access gradient values

fn call(&mut self, input: &[&Tensor]) -> Vec<Tensor>