爱因斯坦表达式操作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