About einsum function in mttr.py

[Hizkai]

Thanks for your generous sharing of this wonderful research results! But I have a problem in reproducing the results.

Here I don't understand how the tensor changes in the einsum function and the reason why to conduct it.

Sorry I don't see any details in the paper, looking forward to your reply