Method forward

beginner_source/basics/optimization_tutorial.py:62–65

(self, x)

        )

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

model = NeuralNetwork()