Class GlobalAvgPool2d

ML/src/python/neuralforge/nn/modules.py:121–126

        return output

class GlobalAvgPool2d(nn.Module):
    def __init__(self):
        super().__init__()
    
    def forward(self, x):
        return x.mean([2, 3])

class GlobalMaxPool2d(nn.Module):
    def __init__(self):