def forward(self, x): x = F.relu(self.bn1(self.fc1(x))) x = F.log_softmax(self.bn2(self.fc2(x))) return x