Einops Extensions

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

Citations

@inproceedings{rogozhnikov2022einops,
  title     = {Einops: Clear and Reliable Tensor Manipulations with Einstein-like Notation},
  author    = {Alex Rogozhnikov},
  booktitle = {International Conference on Learning Representations},
  year      = {2022},
  url       = {https://openreview.net/forum?id=oapKSVM2bcj}
}