Struct Conv2dLayer 

pub struct Conv2dLayer { /* private fields */ }

2D convolution layer (FP32).

Implementations

impl Conv2dLayer

pub fn new( in_channels: usize, out_channels: usize, kernel_size: usize, stride: usize, pad: usize, ) -> Self

pub fn with_algorithm( in_channels: usize, out_channels: usize, kernel_size: usize, stride: usize, pad: usize, algorithm: ConvAlgorithm, ) -> Self

Trait Implementations

impl Layer for Conv2dLayer

fn layer_type(&self) -> LayerType

fn set_input(&mut self, input: Tensor)

fn output(&self) -> &Tensor

fn input_ref(&self) -> &Tensor

fn weights_ref(&self) -> &Tensor

fn bias_ref(&self) -> &Tensor

fn read_weights_bias(&mut self, file: &mut File)

fn fwd(&mut self)

fn print(&self)