Struct Conv2dBuilder

pub struct Conv2dBuilder {
    pub input_channels: i32,
    pub output_channels: i32,
    pub kernel_size: SingleOrPair<i32>,
    pub bias: bool,
    pub stride: SingleOrPair<i32>,
    pub padding: SingleOrPair<i32>,
    pub dilation: SingleOrPair<i32>,
    pub groups: i32,
}

Builder for the Conv2d module.

Fields

input_channels: i32

Number of input channels.

output_channels: i32

Number of output channels.

kernel_size: SingleOrPair<i32>

Size of the convolution filters.

bias: bool

If true, add a learnable bias to the output. Default to Conv2d::DEFAULT_BIAS if not specified.

stride: SingleOrPair<i32>

Stride. Default to Conv2d::DEFAULT_STRIDE if not specified.

padding: SingleOrPair<i32>

Padding. Default to Conv2d::DEFAULT_PADDING if not specified.

dilation: SingleOrPair<i32>

Dilation. Default to Conv2d::DEFAULT_DILATION if not specified.

groups: i32

Groups. Default to Conv2d::DEFAULT_GROUPS if not specified.

Implementations

impl Conv2dBuilder

pub fn new( input_channels: impl Into<i32>, output_channels: impl Into<i32>, kernel_size: impl Into<SingleOrPair<i32>>, ) -> Self

Creates a new Conv2dBuilder.

impl Conv2dBuilder

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

Sets the value of [bias].

pub fn stride(self, stride: impl Into<SingleOrPair<i32>>) -> Self

Sets the value of [stride].

pub fn padding(self, padding: impl Into<SingleOrPair<i32>>) -> Self

Sets the value of [padding].

pub fn dilation(self, dilation: impl Into<SingleOrPair<i32>>) -> Self

Sets the value of [dilation].

pub fn groups(self, groups: impl Into<i32>) -> Self

Sets the value of [groups].

Trait Implementations

impl Builder<Conv2d> for Conv2dBuilder

type Error = Exception

Error with building

fn build(self) -> Result<Conv2d, Self::Error>

Build the type

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.