From 7b9fd63649319194897e196bcb54e00acc2a7d78 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Fri, 18 May 2018 03:32:17 -0700 Subject: [PATCH] Correct commnet --- layers/attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/layers/attention.py b/layers/attention.py index 9c63a85f..f598e182 100644 --- 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