| 12 | import torch |
| 13 | |
| 14 | class Model(torch.nn.Module): |
| 15 | def __init__(self): |
| 16 | super(Model, self).__init__() |
| 17 | self.hid1 = torch.nn.Linear(4, 7) # 4-7-3 |
| 18 | self.oupt = torch.nn.Linear(7, 3) |
| 19 | |
| 20 | def forward(self, x): |
| 21 | z = torch.tanh(self.hid1(x)) |
| 22 | z = self.oupt(z) |
| 23 | return z |
| 24 | |
| 25 | |
| 26 | def parse(args): |