Fix mx missing parenthesis

naviat · Feb 9, 2023 · 046e7d7 · 046e7d7
1 parent d2ab1eb
commit 046e7d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chapter_attention-mechanisms-and-transformers/transformer.md b/chapter_attention-mechanisms-and-transformers/transformer.md
@@ -387,7 +387,7 @@ so that [**the output tensor also has the same shape after the addition operatio
 add_norm = AddNorm(0.5)
 add_norm.initialize()
 shape = (2, 3, 4)
-d2l.check_shape(add_norm(d2l.ones(shape), d2l.ones(shape), shape)
+d2l.check_shape(add_norm(d2l.ones(shape), d2l.ones(shape)), shape)
 ```
 
 ```{.python .input}