最近在看vit_pytorch代码,看到里面有很多地方用到einops来对tensor操作,整理记录一下.
官方:
https://github.com/arogozhnikov/einops
展示:
文档:
1:最好的入门就是官方文档
https://github.com/arogozhnikov/einops/blob/master/docs/1-einops-basics.ipynb
2:知乎的示例
https://zhuanlan.zhihu.com/p/372692913
3:Writing a better code with pytorch and einops
介绍了在pytorch使用einops,来简化网络搭建的一堆示例
http://einops.rocks/pytorch-examples.htmlhttp://einops.rocks/pytorch-examples.html