diff --git a/chapter_appendix-mathematics-for-deep-learning/geometry-linear-algebraic-ops.md b/chapter_appendix-mathematics-for-deep-learning/geometry-linear-algebraic-ops.md
index 312aaa5800..49b41fbe3e 100644
--- a/chapter_appendix-mathematics-for-deep-learning/geometry-linear-algebraic-ops.md
+++ b/chapter_appendix-mathematics-for-deep-learning/geometry-linear-algebraic-ops.md
@@ -1005,7 +1005,7 @@ np.einsum(B, [0, 1, 2], A, [0, 3], v, [1], [2, 3])
 
 ```{.python .input}
 #@tab pytorch
-# PyTorch does not support this type of notation.
+torch.einsum(B, [0, 1, 2], A, [0, 3], v, [1], [2, 3])
 ```
 
 ```{.python .input}