Merge pull request #90 from jmribeiro/patch-1

Replaced .view with .reshape on Attention
PetrochukM · Dec 12, 2019 · 5cc4bb1 · 5cc4bb1
2 parents 46406d3 + ec82cf1
commit 5cc4bb1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchnlp/nn/attention.py b/torchnlp/nn/attention.py
@@ -61,9 +61,9 @@ def forward(self, query, context):
         query_len = context.size(1)
 
         if self.attention_type == "general":
-            query = query.view(batch_size * output_len, dimensions)
+            query = query.reshape(batch_size * output_len, dimensions)
             query = self.linear_in(query)
-            query = query.view(batch_size, output_len, dimensions)
+            query = query.reshape(batch_size, output_len, dimensions)
 
         # TODO: Include mask on PADDING_INDEX?