写 torch / tf 代码时大家有啥记住 size 的技巧或者工具吗，经常各种 view、unsqueeze、transpose 之后绕着绕着就把自己绕晕了

[Morriaty]

或者有啥关键字供搜索吗，我在 google 上尝试搜索 track tensor size ，没有搜到想搜的内容

[pochy06]

每次 reshape 之前都加条注释，写一下 shape 变换 orz
因为模型不是代码量最多的地方，所以多写点注释也不费事儿

[ronpay]

可以试试 einops
https://github.com/arogozhnikov/einops

[Morriaty]

回来自答下，以便后来者能查看。
经过一番查找，目前找到的最符合我初始需求的是[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 ，还在摸索中。

[Morriaty]

tensor_annotation_demo
https://gist.github.com/Morriaty-The-Murderer/c621922429d1f6b4a91ed1fdc97798e3