einops is a flexible tensor operations library using Einstein-inspired notation that works with...