einops provides a set of tensor operations for NumPy, PyTorch, TensorFlow and other frameworks using a readable and reliable notation.

WWW: https://github.com/arogozhnikov/einops
