Struct Conv2d 

pub struct Conv2d { /* private fields */ }

2D Convolution Layer

Performs 2D convolution on NHWC tensors with configurable:

• Kernel size
• Stride
• Padding
• Groups (for depthwise and grouped convolutions)

Kernel layout: [out_channels, kernel_h, kernel_w, in_channels] (OHWI)

Implementations

impl Conv2d

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

Create a new Conv2d layer with Xavier initialization

pub fn builder( in_channels: usize, out_channels: usize, kernel_size: usize, ) -> Conv2dBuilder

Create a Conv2d builder

pub fn with_bias( in_channels: usize, out_channels: usize, kernel_size: usize, stride: usize, padding: usize, ) -> Self

Create Conv2d with bias

pub fn output_shape_nchw(&self, input_shape: &TensorShape) -> TensorShape

Get the output shape for a TensorShape input (NCHW format)

pub fn set_weights(&mut self, weights: Vec<f32>) -> CnnResult<()>

Set the weights directly

pub fn set_bias(&mut self, bias: Vec<f32>) -> CnnResult<()>

Set the bias

pub fn output_shape(&self, input_shape: &[usize]) -> CnnResult<Vec<usize>>

Get the output shape for a given input shape

pub fn weights(&self) -> &[f32]

Get weights reference

pub fn bias(&self) -> Option<&[f32]>

Get bias reference

pub fn kernel_size(&self) -> usize

Get the kernel size

pub fn stride(&self) -> usize

Get the stride

pub fn padding(&self) -> usize

Get the padding

pub fn out_channels(&self) -> usize

Get the number of output channels

pub fn in_channels(&self) -> usize

Get the number of input channels

pub fn groups(&self) -> usize

Get the number of groups

Trait Implementations

impl Clone for Conv2d

fn clone(&self) -> Conv2d

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Conv2d

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

Formats the value using the given formatter.

impl Layer for Conv2d

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

Perform the forward pass

fn name(&self) -> &'static str

Get the layer name

fn num_params(&self) -> usize

Get the number of parameters