fix shape comments (#3025)

allenai · Jul 9, 2019 · 64d16ac · 64d16ac
1 parent 354a19b
commit 64d16ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/allennlp/models/encoder_decoders/simple_seq2seq.py b/allennlp/models/encoder_decoders/simple_seq2seq.py
@@ -441,7 +441,7 @@ def _prepare_attended_input(self,
         """Apply attention over encoder outputs and decoder state."""
         # Ensure mask is also a FloatTensor. Or else the multiplication within
         # attention will complain.
-        # shape: (batch_size, max_input_sequence_length, encoder_output_dim)
+        # shape: (batch_size, max_input_sequence_length)
         encoder_outputs_mask = encoder_outputs_mask.float()
 
         # shape: (batch_size, max_input_sequence_length)