Struct AvgPool2dConfig

pub struct AvgPool2dConfig {
    pub kernel_size: [usize; 2],
    pub strides: [usize; 2],
    pub padding: PaddingConfig2d,
    pub count_include_pad: bool,
}

Configuration to create a 2D avg pooling layer using the init function.

Fields

kernel_size: [usize; 2]

The size of the kernel.

strides: [usize; 2]

The strides.

padding: PaddingConfig2d

The padding configuration.

Warning

Only symmetric padding is currently supported. As such, using Same padding with an even kernel size is not supported as it will not produce the same output size.

count_include_pad: bool

If the padding is counted in the denominator when computing the average.

Implementations

impl AvgPool2dConfig

pub fn new(kernel_size: [usize; 2]) -> AvgPool2dConfig

Create a new instance of the config.

impl AvgPool2dConfig

pub fn with_strides(self, strides: [usize; 2]) -> AvgPool2dConfig

The strides.

pub fn with_padding(self, padding: PaddingConfig2d) -> AvgPool2dConfig

The padding configuration.

pub fn with_count_include_pad(self, count_include_pad: bool) -> AvgPool2dConfig

If the padding is counted in the denominator when computing the average.

impl AvgPool2dConfig

pub fn init(&self) -> AvgPool2d

Initialize a new avg pool 2d module.

Trait Implementations

impl Clone for AvgPool2dConfig

fn clone(&self) -> AvgPool2dConfig

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Config for AvgPool2dConfig

fn save<P>(&self, file: P) -> Result<(), Error>

where P: AsRef<Path>,

Saves the configuration to a file.

fn load<P>(file: P) -> Result<Self, ConfigError>

where P: AsRef<Path>,

Loads the configuration from a file.

fn load_binary(data: &[u8]) -> Result<Self, ConfigError>

Loads the configuration from a binary buffer.

impl Debug for AvgPool2dConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for AvgPool2dConfig

fn deserialize<D>( deserializer: D, ) -> Result<AvgPool2dConfig, <D as Deserializer<'de>>::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for AvgPool2dConfig

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

Formats the value using the given formatter.

impl Serialize for AvgPool2dConfig

fn serialize<S>( &self, serializer: S, ) -> Result<<S as Serializer>::Ok, <S as Serializer>::Error>

where S: Serializer,

Serialize this value into the given Serde serializer.