Struct Linear 

pub struct Linear<T: Num, D: Device> {
    pub weight: Variable<T, D>,
    pub bias: Option<Variable<T, D>>,
    /* private fields */
}

Fields

weight: Variable<T, D>

bias: Option<Variable<T, D>>

Implementations

impl<T: Num, D: Device> Linear<T, D>

pub fn new(in_features: usize, out_features: usize, use_bias: bool) -> Self

where StandardNormal: Distribution<T>,

pub fn to<Dout: Device>(self) -> Linear<T, Dout>

Trait Implementations

impl<T: Num, D: Device> Module<T, D> for Linear<T, D>

type Input = Variable<T, D>

type Output = Variable<T, D>

fn call(&self, input: Variable<T, D>) -> Variable<T, D>

impl<T: Num, D: Device> Parameters<T, D> for Linear<T, D>

fn weights(&self) -> HashMap<String, Variable<T, D>>

fn biases(&self) -> HashMap<String, Variable<T, D>>

fn parameters(&self) -> HashMap<String, Variable<T, D>>

fn load_parameters(&mut self, parameters: HashMap<String, Variable<T, D>>)