We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 732b445 commit de6d396Copy full SHA for de6d396
1 file changed
onmt/modules/multi_headed_attn.py
@@ -86,8 +86,8 @@ def forward(self, key, value, query, mask=None,
86
value vectors ``(batch, key_len, dim)``
87
query (FloatTensor): set of `query_len`
88
query vectors ``(batch, query_len, dim)``
89
- mask: binary mask indicating which keys have
90
- non-zero attention ``(batch, query_len, key_len)``
+ mask: binary mask 1/0 indicating which keys have
+ zero / non-zero attention ``(batch, query_len, key_len)``
91
Returns:
92
(FloatTensor, FloatTensor):
93
0 commit comments