Method __init__

ML/src/python/neuralforge/nn/modules.py:129–130

(self)

class GlobalMaxPool2d(nn.Module):
    def __init__(self):
        super().__init__()
    
    def forward(self, x):
        return x.max(dim=2)[0].max(dim=2)[0]