 |
ronpayV2EX 第 501241 号会员，加入于 2020-07-28 20:33:58 +08:00 |
ronpay 最近回复了
2023-12-25 23:01:13 +08:00 回复了 Morriaty 创建的主题 › 程序员 › 写 torch / tf 代码时大家有啥记住 size 的技巧或者工具吗，经常各种 view、unsqueeze、transpose 之后绕着绕着就把自己绕晕了 |
可以试试 einops
https://github.com/arogozhnikov/einops
https://github.com/arogozhnikov/einops
Select Language