search

arogozhnikov / einops

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

Combine rearrange and repeat? #344

Closed bjourne closed 1 month ago

bjourne commented 1 month ago

I asked ChatGPT about it and it said:

combine_repeat

I think it would be cool if it worked. :)

arogozhnikov commented 1 month ago

wrong prompt - wrong answer

Function exists since 2018. https://github.com/arogozhnikov/einops/blob/main/docs/1-einops-basics.ipynb