einops 爱因斯坦表达式

 @halomaster 18 Sep 2022, 12:54 GMT【AI】 

爱因斯坦表达式操作tensor有两个好处: 1. 人类理解简单; 2. 编译器/JIT理解简单,方便生成GPU/TPU等加速器的指令; [【einops】](https://zhuanlan.zhihu.com/p/372692913) [colab](https://drive.google.com/file/d/1sT3AQlf1Khw6iYoWjG9CgHbwB_NAJnsj/view?usp=sharing) https://gist.github.com/linkerlin/229871bb727dca0122752a499efdab25

0 replies