Method __init__

ML/src/python/neuralforge/nn/modules.py:136–139

(self)

class AdaptiveAvgMaxPool2d(nn.Module):
    def __init__(self):
        super().__init__()
        self.avg_pool = GlobalAvgPool2d()
        self.max_pool = GlobalMaxPool2d()
    
    def forward(self, x):
        avg = self.avg_pool(x)