Method __init__

ML/src/python/neuralforge/nn/modules.py:122–123

(self)

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