写 torch / tf 代码时大家有啥记住 size 的技巧或者工具吗，经常各种 view、unsqueeze、transpose 之后绕着绕着就把自己绕晕了
Morriaty · 342 天前 · 1069 次点击这是一个创建于 342 天前的主题，其中的信息可能已经有所发展或是发生改变。
或者有啥关键字供搜索吗，我在 google 上尝试搜索 track tensor size ，没有搜到想搜的内容
 |
1
pochy06 342 天前  1
每次 reshape 之前都加条注释，写一下 shape 变换 orz
因为模型不是代码量最多的地方，所以多写点注释也不费事儿 |
 |
2
ronpay 339 天前
可以试试 einops
https://github.com/arogozhnikov/einops |
 |
3
Morriaty OP 回来自答下，以便后来者能查看。
经过一番查找，目前找到的最符合我初始需求的是[tsalib]( https://github.com/ofnote/tsalib) 例子如下 <script src="https://gist.github.com/Morriaty-The-Murderer/c621922429d1f6b4a91ed1fdc97798e3.js"></script> 其中也结合了 2L 提到的`einops`，其实`tsalib`本身是自带类似的`warp`操作的，但由于`tsalib`本身很久没维护了（ last commit 4years ago ），所以操作 tensor 还是选择了目前稳定维护的`einops`，仅保留了`tsalib`的`type annotation` 由于我也是刚开始使用，所以不太清楚这样子写是否是 best practise ，还在摸索中。 |
|
4
Morriaty OP tensor_annotation_demo
https://gist.github.com/Morriaty-The-Murderer/c621922429d1f6b4a91ed1fdc97798e3 |
Select Language