Correct commnet

2018-05-18 03:32:17 -07:00 · 2018-05-18 03:32:17 -07:00 · 7b9fd63649
parent d43b4ccc1e
commit 7b9fd63649
1 changed files with 1 additions and 1 deletions
--- a/layers/attention.py
+++ b/layers/attention.py
@ -13,8 +13,8 @@ class BahdanauAttention(nn.Module):
    def forward(self, annots, query):
        """
        Shapes:
-            - query: (batch, 1, dim) or (batch, dim)
            - annots: (batch, max_time, dim)
+            - query: (batch, 1, dim) or (batch, dim)
        """
        if query.dim() == 2:
            # insert time-axis for broadcasting