Eistein Summation

Eistein Summation is very convenient for high dimension matrix multiplication.

In pytorch, it can be used as:

torch.einsum(‘ijk,li->ljk’, Mat_a, Mat_b)