bug fix argparser

2026-09-10 11:50:20 +08:00 · 2019-10-31 15:13:39 +01:00
parent 9b6021318d
commit ec579d02a1
3 changed files with 10 additions and 3 deletions
@@ -226,6 +226,13 @@ class Attention(nn.Module):
        return alpha

    def forward(self, query, inputs, processed_inputs, mask):
+        """
+        shapes:
+            query: B x D_attn_rnn
+            inputs: B x T_en x D_en
+            processed_inputs:: B x T_en x D_attn
+            mask: B x T_en
+        """
        if self.location_attention:
            attention, _ = self.get_location_attention(
                query, processed_inputs)