Not hard to understand. The visualization is more or less the computation graph that PyTorch builds up. And the einsum code is even clearer. There’s definitely

Not hard to understand. The visualization is more or less the computation graph that PyTorch builds up. And the einsum code is even clearer. There’s definitely a practice effect though. I know people who aren’t used to it will have their eyes glaze over when they read einsum notation.