Method forward

demo.py:21–22

(self, x)

            nn.Linear(h2, 1)
        )
    def forward(self, x):
        return self.net(x)


device = torch.device("cuda" if torch.cuda.is_available() else "cpu")