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.
Flattens a contiguous range of dims into a tensor. |
|
Unflatten a tensor dim expanding it to a desired shape. |
|
Applies a 1D average pooling over an input signal composed of several input planes. |
|
Applies a 2D average pooling over an input signal composed of several input planes. |
|
Applies a 3D average pooling over an input signal composed of several input planes. |
|
Applies a 1D max pooling over an input signal composed of several input planes. |
|
Applies a 2D max pooling over an input signal composed of several input planes. |
|
Applies a 3D max pooling over an input signal composed of several input planes. |
|
Computes a partial inverse of MaxPool1d. |
|
Computes a partial inverse of MaxPool2d. |
|
Computes a partial inverse of MaxPool3d. |
|
Applies a 1D power-average pooling over an input signal composed of several input planes. |
|
Applies a 2D power-average pooling over an input signal composed of several input planes. |
|
Applies a 3D power-average pooling over an input signal composed of several input planes. |
|
Applies a 1D adaptive average pooling over an input signal composed of several input planes. |
|
Applies a 2D adaptive average pooling over an input signal composed of several input planes. |
|
Applies a 3D adaptive average pooling over an input signal composed of several input planes. |
|
Applies a 1D adaptive max pooling over an input signal composed of several input planes. |
|
Applies a 2D adaptive max pooling over an input signal composed of several input planes. |
|
Applies a 3D adaptive max pooling over an input signal composed of several input planes. |