From 145de87b4cf396491bbe7e75b7e69f281138560f Mon Sep 17 00:00:00 2001 From: BanananaFish <43786104+BanananaFish@users.noreply.github.com> Date: Sun, 12 May 2024 13:25:12 +0800 Subject: [PATCH] Update torch.einsum sublist format example torch 1.10 updated sublist format in torch.einsum api, see https://pytorch.org/docs/stable/generated/torch.einsum.html --- .../geometry-linear-algebraic-ops.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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}