def __init__(self, hid):
super(Net, self).__init__()
self.linear1 = torch.nn.Linear(2, hid)
self.linear2 = torch.nn.Linear(hid, hid)
self.linear3 = torch.nn.Linear(hid, 1)
def forward(self,hid):
linear1 = self.linear1(hid)
self.hid1 = torch.tan(linear1)
linear2 = self.linear2(self.hid1)
self.hid2 = torch.tan(linear2)
linear3 = self.linear3(self.hid2)
self.output = torch.sigmoid(linear3)
return self.output