NNLM-Torch.py nn.functional to torch

baichenjing · Feb 20, 2019 · 3765bb4 · 3765bb4
1 parent 06ca9dd
commit 3765bb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/1-1.NNLM/NNLM-Torch.py b/1-1.NNLM/NNLM-Torch.py
@@ -46,7 +46,7 @@ def __init__(self):
 
     def forward(self, X):
         input = X.view(-1, n_step * n_class) # [batch_size, n_step * n_class]
-        tanh = nn.functional.tanh(self.d + torch.mm(input, self.H)) # [batch_size, n_hidden]
+        tanh = torch.tanh(self.d + torch.mm(input, self.H)) # [batch_size, n_hidden]
         output = self.b + torch.mm(input, self.W) + torch.mm(tanh, self.U) # [batch_size, n_class]
         return output