Pooling Layers#

Module

MAdds

FLOPs

Parameters

Memory

torch.nn.MaxPool1d

x

x

NA

x

torch.nn.MaxPool2d

✓

✓

NA

✓

torch.nn.MaxPool3d

x

x

NA

x

torch.nn.MaxUnpool1d

x

x

NA

x

torch.nn.MaxUnpool2d

x

x

NA

x

torch.nn.MaxUnpool3d

x

x

NA

x

torch.nn.AvgPool1d

x

x

NA

x

torch.nn.AvgPool2d

✓

✓

NA

✓

torch.nn.AvgPool3d

x

x

NA

x

torch.nn.FractionalMaxPool2d

x

x

NA

x

torch.nn.FractionalMaxPool3d

x

x

NA

x

torch.nn.LPPool1d

x

x

NA

x

torch.nn.LPPool2d

x

x

NA

x

torch.nn.LPPool3d

x

x

NA

x

torch.nn.AdaptiveMaxPool1d

x

x

NA

x

torch.nn.AdaptiveMaxPool2d

x

x

NA

x

torch.nn.AdaptiveMaxPool3d

x

x

NA

x

torch.nn.AdaptiveAvgPool1d

x

x

NA

x

torch.nn.AdaptiveAvgPool2d

x

x

NA

x

torch.nn.AdaptiveAvgPool3d

x

x

NA

x