Struct Conv2DBuilder 

pub struct Conv2DBuilder { /* private fields */ }

Builder for creating Conv2D layers

Implementations

impl Conv2DBuilder

pub fn new(name: &str) -> Self

Create a new Conv2D builder

pub const fn filter_size(self, height: usize, width: usize) -> Self

Set filter dimensions

pub const fn channels(self, input: usize, output: usize) -> Self

Set input and output channels

pub fn stride(self, stride: (usize, usize)) -> Self

Set stride

pub fn padding(self, padding: (usize, usize)) -> Self

Set padding

pub fn withbias(self, withbias: bool) -> Self

Set whether to include bias

pub fn activation(self, activation: ActivationFunc) -> Self

Set activation function

pub fn build(self) -> Conv2D

Build the Conv2D layer