Struct Conv2DLayer 

pub struct Conv2DLayer {
    pub filters: Tensor,
    pub bias: Tensor,
    pub stride: usize,
    pub padding: usize,
}

Fields

filters: Tensor

bias: Tensor

stride: usize

padding: usize

Implementations

impl Conv2DLayer

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

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

pub fn parameter_count(&self) -> usize

Trait Implementations

impl Clone for Conv2DLayer

fn clone(&self) -> Conv2DLayer

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Conv2DLayer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.