fix output shape comment.

HarshTrivedi · Feb 9, 2019 · db96a0b · db96a0b
1 parent 5622019
commit db96a0b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -211,7 +211,7 @@ output, input_sizes = pad_packed_sequence(packed_output, batch_first=True)
 #                            [ 0.          0.          0.          0.          0.        ]   <pad>
 #                            [ 0.          0.          0.          0.          0.        ]   <pad>
 #                            [ 0.          0.          0.          0.          0.        ]]] <pad>
-# output.shape : ( batch_size X max_seq_len X hidden_dim) = (3 X 18 X 4)
+# output.shape : ( batch_size X max_seq_len X hidden_dim) = (3 X 8 X 5)
 
 # Or if you just want the final hidden state?
 print(ht[-1])

diff --git a/pad_packed_demo.py b/pad_packed_demo.py
@@ -214,7 +214,7 @@
 #                            [ 0.          0.          0.          0.          0.        ]   <pad>
 #                            [ 0.          0.          0.          0.          0.        ]   <pad>
 #                            [ 0.          0.          0.          0.          0.        ]]] <pad>
-# output.shape : ( batch_size X max_seq_len X hidden_dim) = (3 X 18 X 4)
+# output.shape : ( batch_size X max_seq_len X hidden_dim) = (3 X 8 X 5)
 
 # Or if you just want the final hidden state?
 print(ht[-1])