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.