From 3765bb4eebd6008bd08e35170a0daceb14a6267a Mon Sep 17 00:00:00 2001
From: us <rahmetsaritekin@gmail.com>
Date: Wed, 20 Feb 2019 10:50:44 +0300
Subject: [PATCH] NNLM-Torch.py `nn.functional` to `torch`
---
1-1.NNLM/NNLM-Torch.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/1-1.NNLM/NNLM-Torch.py b/1-1.NNLM/NNLM-Torch.py
index baaafce..9905040 100644
--- 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