Struct burn::nn::pool::AvgPool2d

pub struct AvgPool2d {
    pub stride: [usize; 2],
    pub kernel_size: [usize; 2],
    pub padding: Ignored<PaddingConfig2d>,
    pub count_include_pad: bool,
}

Applies a 2D avg pooling over input tensors.

Should be created with AvgPool2dConfig.

Remarks

The zero-padding values will be included in the calculation of the average. This means that the zeros are counted as legitimate values, and they contribute to the denominator when calculating the average. This is equivalent to torch.nn.AvgPool2d with count_include_pad=True.

TODO: Add support for count_include_pad=False, see Issue 636

Fields

stride: [usize; 2]

Stride of the pooling.

kernel_size: [usize; 2]

Size of the kernel.

padding: Ignored<PaddingConfig2d>

Padding configuration.

count_include_pad: bool

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

Implementations

impl AvgPool2d

pub fn forward<B>(&self, input: Tensor<B, 4>) -> Tensor<B, 4>

where B: Backend,

Applies the forward pass on the input tensor.

See avg_pool2d for more information.

Shapes

• input: [batch_size, channels, height_in, width_in]
• output: [batch_size, channels, height_out, width_out]

Trait Implementations

impl<B> AutodiffModule<B> for AvgPool2d

where B: AutodiffBackend,

type InnerModule = AvgPool2d

Inner module without auto-differentiation.

fn valid(&self) -> <AvgPool2d as AutodiffModule<B>>::InnerModule

Get the same module, but on the inner backend without auto-differentiation.

impl Clone for AvgPool2d

fn clone(&self) -> AvgPool2d

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for AvgPool2d

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

Formats the value using the given formatter.

impl Display for AvgPool2d

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

Formats the value using the given formatter.

impl<B> Module<B> for AvgPool2d

where B: Backend,

type Record = ConstantRecord

Type to save and load the module.

fn visit<V>(&self, _visitor: &mut V)

where V: ModuleVisitor<B>,

Visit each tensor parameter in the module with a visitor.

fn map<M>(self, _mapper: &mut M) -> AvgPool2d

where M: ModuleMapper<B>,

Map each tensor parameter in the module with a mapper.

fn load_record(self, _record: <AvgPool2d as Module<B>>::Record) -> AvgPool2d

Load the module state from a record.

fn into_record(self) -> <AvgPool2d as Module<B>>::Record

Convert the module into a record containing the state.

fn to_device(self, _: &<B as Backend>::Device) -> AvgPool2d

Move the module and all of its sub-modules to the given device.

fn fork(self, _: &<B as Backend>::Device) -> AvgPool2d

Fork the module and all of its sub-modules to the given device.

fn collect_devices( &self, devices: Vec<<B as Backend>::Device>, ) -> Vec<<B as Backend>::Device>

Return all the devices found in the underneath module tree added to the given vector without duplicates.

fn devices(&self) -> Vec<<B as Backend>::Device>

Return all the devices found in the underneath module tree without duplicates.

fn no_grad(self) -> Self

Each tensor in the module tree will not require grad.

fn num_params(&self) -> usize

Get the number of parameters the module has, including all of its sub-modules.

fn save_file<FR, PB>( self, file_path: PB, recorder: &FR, ) -> Result<(), RecorderError>

where FR: FileRecorder<B>, PB: Into<PathBuf>,

Save the module to a file using the provided file recorder.

fn load_file<FR, PB>( self, file_path: PB, recorder: &FR, device: &<B as Backend>::Device, ) -> Result<Self, RecorderError>

where FR: FileRecorder<B>, PB: Into<PathBuf>,

Load the module from a file using the provided file recorder.

fn quantize_weights<C>(self, quantizer: &mut Quantizer<C>) -> Self

where C: Calibration,

Quantize the weights of the module.

impl ModuleDisplay for AvgPool2d

fn custom_settings(&self) -> Option<DisplaySettings>

Custom display settings for the module.

fn custom_content(&self, content: Content) -> Option<Content>

Custom attributes for the module.

fn format(&self, passed_settings: DisplaySettings) -> String

Formats the module with provided display settings.

impl ModuleDisplayDefault for AvgPool2d

fn content(&self, content: Content) -> Option<Content>

Attributes of the module used for display purposes.

fn num_params(&self) -> usize

Gets the number of the parameters of the module.