Skip to content

GitLab

Commit 4cd908ef authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Fix typo. 

PiperOrigin-RevId: 315738983
parent 99bdc3dc
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/modeling/layers/attention.py
View file @ 4cd908ef
...@@ -145,7 +145,7 @@ class MultiHeadAttention(tf.keras.layers.Layer): ...@@ -145,7 +145,7 @@ class MultiHeadAttention(tf.keras.layers.Layer):
>>> target = tf.keras.Input(shape=[8, 16]) >>> target = tf.keras.Input(shape=[8, 16])
>>> source = tf.keras.Input(shape=[4, 16]) >>> source = tf.keras.Input(shape=[4, 16])
>>> mask_tensor = tf.keras.Input(shape=[8, 4]) >>> mask_tensor = tf.keras.Input(shape=[8, 4])
>>> output_tensor, weights = layer([input_tensor, input_tensor]) >>> output_tensor, weights = layer([target, source])
>>> print(output_tensor.shape), print(weights.shape) >>> print(output_tensor.shape), print(weights.shape)
(None, 8, 16) (None, 2, 8, 4) (None, 8, 16) (None, 2, 8, 4)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment