Struct Conv2d 

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

Fields

filter: Variable<T, D>

bias: Option<Variable<T, D>>

Implementations

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

pub fn new( input_channel: usize, output_channel: usize, kernel_size: (usize, usize), stride: (usize, usize), padding: (usize, usize), bias: bool, ) -> Self

where StandardNormal: Distribution<T>,

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

Trait Implementations

impl<T: Num, D: Device> Module<T, D> for Conv2d<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 Conv2d<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>>)