Struct Linear 

pub struct Linear { /* private fields */ }

Implementations

impl Linear

pub fn new( in_features: usize, out_features: usize, bias: bool, device: Device, ) -> Self

Trait Implementations

impl Module for Linear

fn forward(&mut self, input: &Tensor) -> Tensor

fn parameters(&self) -> HashMap<String, Arc<RwLock<Tensor>>>

fn train(&mut self)

fn eval(&mut self)

fn zero_grad(&mut self)

fn visit_parameters(&self, f: &mut dyn FnMut(&str, &Tensor))

Visit all parameters with a callback function

fn print_parameters(&self, values: bool)

Print all parameters and their shapes