Skip to content

pooling

Module: pooling.py

This module implements pooling layers for Neural Networks

Authors
  • Lokesh Mohanty (lokeshm@iisc.ac.in)
Version Info
  • 06/01/2025: Initial version

AdaptiveAvgPool1d

Bases: AdaptiveAvgPool1d

Performs a 1D adaptive average pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
89
90
91
92
class AdaptiveAvgPool1d(eqx.nn.AdaptiveAvgPool1d):
    """
    Performs a 1D adaptive average pooling operation
    """

AdaptiveAvgPool2d

Bases: AdaptiveAvgPool2d

Performs a 2D adaptive average pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
95
96
97
98
class AdaptiveAvgPool2d(eqx.nn.AdaptiveAvgPool2d):
    """
    Performs a 2D adaptive average pooling operation
    """

AdaptiveAvgPool3d

Bases: AdaptiveAvgPool3d

Performs a 3D adaptive average pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
101
102
103
104
class AdaptiveAvgPool3d(eqx.nn.AdaptiveAvgPool3d):
    """
    Performs a 3D adaptive average pooling operation
    """

AdaptiveMaxPool1d

Bases: AdaptiveMaxPool1d

Performs a 1D adaptive max pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
107
108
109
110
class AdaptiveMaxPool1d(eqx.nn.AdaptiveMaxPool1d):
    """
    Performs a 1D adaptive max pooling operation
    """

AdaptiveMaxPool2d

Bases: AdaptiveMaxPool2d

Performs a 2D adaptive max pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
113
114
115
116
class AdaptiveMaxPool2d(eqx.nn.AdaptiveMaxPool2d):
    """
    Performs a 2D adaptive max pooling operation
    """

AdaptiveMaxPool3d

Bases: AdaptiveMaxPool3d

Performs a 3D adaptive max pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
119
120
121
122
class AdaptiveMaxPool3d(eqx.nn.AdaptiveMaxPool3d):
    """
    Performs a 3D adaptive max pooling operation
    """

AdaptivePool

Bases: AdaptivePool

Performs an adaptive pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
83
84
85
86
class AdaptivePool(eqx.nn.AdaptivePool):
    """
    Performs an adaptive pooling operation
    """

AvgPool1d

Bases: AvgPool1d

Performs a 1D average pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
47
48
49
50
class AvgPool1d(eqx.nn.AvgPool1d):
    """
    Performs a 1D average pooling operation
    """

AvgPool2d

Bases: AvgPool2d

Performs a 2D average pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
53
54
55
56
class AvgPool2d(eqx.nn.AvgPool2d):
    """
    Performs a 2D average pooling operation
    """

AvgPool3d

Bases: AvgPool3d

Performs a 3D average pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
59
60
61
62
class AvgPool3d(eqx.nn.AvgPool3d):
    """
    Performs a 3D average pooling operation
    """

MaxPool1d

Bases: MaxPool1d

Performs a 1D max pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
65
66
67
68
class MaxPool1d(eqx.nn.MaxPool1d):
    """
    Performs a 1D max pooling operation
    """

MaxPool2d

Bases: MaxPool2d

Performs a 2D max pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
71
72
73
74
class MaxPool2d(eqx.nn.MaxPool2d):
    """
    Performs a 2D max pooling operation
    """

MaxPool3d

Bases: MaxPool3d

Performs a 3D max pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
77
78
79
80
class MaxPool3d(eqx.nn.MaxPool3d):
    """
    Performs a 3D max pooling operation
    """

Pool

Bases: Pool

Performs a pooling operation

Source code in scirex/core/dl/nn/layers/pooling.py 
41
42
43
44
class Pool(eqx.nn.Pool):
    """
    Performs a pooling operation
    """