Method forward

models/classifiers/classifier.py:15–16

(self, input)

class Flatten(nn.Module):
    def forward(self, input):
        return input.view(input.size(0), -1)


class Mnist_CNN(nn.Module):