torchadf.nn.modules.pooling.AvgPool3d

class torchadf.nn.modules.pooling.AvgPool3d(kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True, divisor_override=None, mode='diag')

A 3D average pooling layer.

Assumed Density Filtering (ADF) version of torch.nn.AvgPool3d.

Parameters:

kernel_sizeint or tuple of int

Size of the pooling window.

strideint or tuple of int, optional

Stride of the pooling window (Default = kernel_size).

paddingint or tuple of int, optional

Implicit zero padding on both sides of input (Default 0).

ceil_modebool, optional

Use ceil instead of floor to compute output shape (Default False).

count_include_padbool, optional

Include zero-padding in average calculations (Default True).

divisor_overrideoptional,

Will be used as divisor if specified, otherwise kernel_size will be used (Default None).

mode{“diag”, “diagonal”, “lowrank”, “half”, “full”}, optional

Covariance propagation mode (Default “diag”).