arogozhnikov / einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
https://einops.rocks
MIT License
8.41k stars 350 forks source link

Uniform support for list inputs across rearrange/reduce/repeat #263

Closed arogozhnikov closed 1 year ago