Update torch.einsum sublist format example

d2l-ai / d2l-en

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

https://D2L.ai

Other

22.45k stars 4.19k forks source link

Update torch.einsum sublist format example #2597

Open BanananaFish opened 1 month ago

BanananaFish commented 1 month ago

Description of changes: torch 1.10 updated sublist format in torch.einsum api, see https://pytorch.org/docs/stable/generated/torch.einsum.html

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.