Method forward

beginner_source/basics/quickstart_tutorial.py:106–109

(self, x)

        )

    def forward(self, x):
        x = self.flatten(x)
        logits = self.linear_relu_stack(x)
        return logits

model = NeuralNetwork().to(device)
print(model)