Writing code with einsum, and knowing what each index represents like time, space, element is so much more intuitive then trying to reason about just how need to transpose things so that everything works to do the tensor contraction you want.

I will be studying this