The einops library provides a powerful and flexible way to manipulate tensors using the Einstein summation convention. The
einops uses a different convention than the traditional one. In particular, the axis names
can be more than one letter long and are separated by spaces.