Method __init__

ML/src/python/neuralforge/nn/modules.py:147–149

(self, start_dim=1)

class Flatten(nn.Module):
    def __init__(self, start_dim=1):
        super().__init__()
        self.start_dim = start_dim
    
    def forward(self, x):
        return x.flatten(self.start_dim)