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 |