einops

Deep learning operations reinvented (for pytorch, tensorflow, chainer, gluon and others)