Pooling and Reshaping

Pooling and Reshaping#

Downsampling, upsampling, and shape manipulation operations for spatial data. Includes average pooling, max pooling, Lp-norm pooling, unpooling for reconstruction, and adaptive pooling for fixed output sizes. Flatten and Unflatten enable seamless transitions between spatial and flat representations.

Flatten

Flattens a contiguous range of dims into a tensor.

Unflatten

Unflatten a tensor dim expanding it to a desired shape.

AvgPool1d

Applies a 1D average pooling over an input signal composed of several input planes.

AvgPool2d

Applies a 2D average pooling over an input signal composed of several input planes.

AvgPool3d

Applies a 3D average pooling over an input signal composed of several input planes.

MaxPool1d

Applies a 1D max pooling over an input signal composed of several input planes.

MaxPool2d

Applies a 2D max pooling over an input signal composed of several input planes.

MaxPool3d

Applies a 3D max pooling over an input signal composed of several input planes.

MaxUnpool1d

Computes a partial inverse of MaxPool1d.

MaxUnpool2d

Computes a partial inverse of MaxPool2d.

MaxUnpool3d

Computes a partial inverse of MaxPool3d.

LPPool1d

Applies a 1D power-average pooling over an input signal composed of several input planes.

LPPool2d

Applies a 2D power-average pooling over an input signal composed of several input planes.

LPPool3d

Applies a 3D power-average pooling over an input signal composed of several input planes.

AdaptiveAvgPool1d

Applies a 1D adaptive average pooling over an input signal composed of several input planes.

AdaptiveAvgPool2d

Applies a 2D adaptive average pooling over an input signal composed of several input planes.

AdaptiveAvgPool3d

Applies a 3D adaptive average pooling over an input signal composed of several input planes.

AdaptiveMaxPool1d

Applies a 1D adaptive max pooling over an input signal composed of several input planes.

AdaptiveMaxPool2d

Applies a 2D adaptive max pooling over an input signal composed of several input planes.

AdaptiveMaxPool3d

Applies a 3D adaptive max pooling over an input signal composed of several input planes.