fix bi-LSTM hidden and cell state shape comments

Yuhuishishishi · Yuhuishishishi · commit 90dc12a3d4ba · 2020-05-25T00:04:14.000-07:00
diff --git a/3-3.Bi-LSTM/Bi-LSTM-Torch.py b/3-3.Bi-LSTM/Bi-LSTM-Torch.py
@@ -47,8 +47,8 @@ def __init__(self):
     def forward(self, X):
         input = X.transpose(0, 1)  # input : [n_step, batch_size, n_class]
 
-        hidden_state = Variable(torch.zeros(1*2, len(X), n_hidden))   # [num_layers(=1) * num_directions(=1), batch_size, n_hidden]
-        cell_state = Variable(torch.zeros(1*2, len(X), n_hidden))     # [num_layers(=1) * num_directions(=1), batch_size, n_hidden]
+        hidden_state = Variable(torch.zeros(1*2, len(X), n_hidden))   # [num_layers(=1) * num_directions(=2), batch_size, n_hidden]
+        cell_state = Variable(torch.zeros(1*2, len(X), n_hidden))     # [num_layers(=1) * num_directions(=2), batch_size, n_hidden]
 
         outputs, (_, _) = self.lstm(input, (hidden_state, cell_state))
         outputs = outputs[-1]  # [batch_size, n_hidden]
@@ -75,4 +75,4 @@ def forward(self, X):
 
 predict = model(input_batch).data.max(1, keepdim=True)[1]
 print(sentence)
-print([number_dict[n.item()] for n in predict.squeeze()])
+print([number_dict[n.item()] for n in predict.squeeze()])
