Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class net(nn.Module):
- def __init__(self):
- super(net, self).__init__()
- self.ln1 = nn.Linear(11,32,bias=False)
- self.ln2 = nn.Linear(32,32,bias=False)
- self.ln3 = nn.Linear(32,32,bias=False)
- self.ln4 = nn.Linear(32,1,bias=False)
- nninit.uniform(self.ln1.weight, a=-1,b=1)
- nninit.uniform(self.ln2.weight,a=-1,b=1)
- nninit.uniform(self.ln3.weight,a=-1,b=1)
- nninit.uniform(self.ln4.weight,a=-1,b=1)
- self.tanh1 = nn.Tanh()
- self.tanh2 = nn.Tanh()
- self.tanh3 = nn.Tanh()
- self.sigmoid = nn.Sigmoid()
- def forward(self,x):
- U = self.ln1(x)
- U = self.tanh1(U)
- U = self.ln2(U)
- U = self.tanh2(U)
- U = self.ln3(U)
- U = self.tanh3(U)
- U = torch.sin(self.ln4(U))
- return self.sigmoid(U)
- G = net()
Add Comment
Please, Sign In to add comment