Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
