Fixes a typo.

PiperOrigin-RevId: 311196489

Fixes a typo.
PiperOrigin-RevId: 311196489
3ae3aecf · A. Unique TensorFlower · 8518d053 · 3ae3aecf
Commit 3ae3aecf authored May 12, 2020 by A. Unique TensorFlower
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/nlp/modeling/layers/attention.py official/nlp/modeling/layers/attention.py +1 -1

No files found.
--- a/official/nlp/modeling/layers/attention.py
+++ b/official/nlp/modeling/layers/attention.py
@@ -307,7 +307,7 @@ class MultiHeadAttention(tf.keras.layers.Layer):
        attention to certain positions.

    Returns:
-      attention_output: The result of the computation, of shape [B, F, N, V] or
+      attention_output: The result of the computation, of shape [B, T, N, V] or
        [B, F, E], where `N` is the number of heads and `E` is the query input
        last dimension.
    """