Method forward

Adversarial_AutoEncoder/net/Encoder.py:23–26

(self,x)

            torch.nn.Linear(in_features=256, out_features=out_features),
        )
    def forward(self,x):
        x = x.view(-1,784)
        out = self.encoder(x)
        return out

if __name__ == '__main__':
    model = Encoder(in_features=784,out_features=128)