https://techtalk.digitalpress.blog/decoding-einsum-with-python/