Struct AvgPool1dConfig 

pub struct AvgPool1dConfig {
    pub kernel_size: usize,
    pub stride: usize,
    pub padding: PaddingConfig1d,
    pub count_include_pad: bool,
    pub ceil_mode: bool,
}

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

Fields

kernel_size: usize

The size of the kernel.

stride: usize

The stride.

padding: PaddingConfig1d

The padding configuration.

Supports symmetric and asymmetric padding. Same padding with even kernel sizes will automatically use asymmetric padding to preserve input dimensions.

count_include_pad: bool

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

ceil_mode: bool

If true, use ceiling instead of floor for output size calculation.

Implementations

impl AvgPool1dConfig

pub fn new(kernel_size: usize) -> AvgPool1dConfig

Create a new instance of the config.

Arguments

Required Arguments

kernel_size

The size of the kernel.

Default Arguments

stride

The stride.

• Defaults to "kernel_size"

padding

The padding configuration.

Supports symmetric and asymmetric padding. Same padding with even kernel sizes will automatically use asymmetric padding to preserve input dimensions.

• Defaults to "PaddingConfig1d::Valid"

count_include_pad

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

• Defaults to "true"

ceil_mode

If true, use ceiling instead of floor for output size calculation.

• Defaults to "false"

impl AvgPool1dConfig

pub fn with_stride(self, stride: usize) -> AvgPool1dConfig

Sets the value for the field stride.

The stride.

• Defaults to "kernel_size"

pub fn with_padding(self, padding: PaddingConfig1d) -> AvgPool1dConfig

Sets the value for the field padding.

The padding configuration.

Supports symmetric and asymmetric padding. Same padding with even kernel sizes will automatically use asymmetric padding to preserve input dimensions.

• Defaults to "PaddingConfig1d::Valid"

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

Sets the value for the field count_include_pad.

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

• Defaults to "true"

pub fn with_ceil_mode(self, ceil_mode: bool) -> AvgPool1dConfig

Sets the value for the field ceil_mode.

If true, use ceiling instead of floor for output size calculation.

• Defaults to "false"

impl AvgPool1dConfig

pub fn init(&self) -> AvgPool1d

Initialize a new avg pool 1d module.

Trait Implementations

impl Clone for AvgPool1dConfig

fn clone(&self) -> AvgPool1dConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Config for AvgPool1dConfig

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 AvgPool1dConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for AvgPool1dConfig

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

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for AvgPool1dConfig

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

Formats the value using the given formatter.

impl Serialize for AvgPool1dConfig

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.