[Embedding] Forward given padding_index param to embedding() (#2504)

allenai · Jul 19, 2019 · 88a61e1 · 88a61e1
1 parent 9ed9e2c
commit 88a61e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/allennlp/modules/token_embedders/embedding.py b/allennlp/modules/token_embedders/embedding.py
@@ -137,6 +137,7 @@ def forward(self, inputs):  # pylint: disable=arguments-differ
         inputs = util.combine_initial_dims(inputs)
 
         embedded = embedding(inputs, self.weight,
+                             padding_idx=self.padding_index,
                              max_norm=self.max_norm,
                              norm_type=self.norm_type,
                              scale_grad_by_freq=self.scale_grad_by_freq,