Struct Conv2dBuilder 

pub struct Conv2dBuilder { /* private fields */ }

Builder for Conv2d layer

Implementations

impl Conv2dBuilder

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

Create a new builder

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

Set stride

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

Set padding

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

Set groups for grouped convolution

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

Set whether to use bias

pub fn build(self) -> CnnResult<Conv2d>

Build the Conv2d layer

Trait Implementations

impl Clone for Conv2dBuilder

fn clone(&self) -> Conv2dBuilder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Conv2dBuilder

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

Formats the value using the given formatter.