最近在调试pytorch代码的时候遇到如下问题,由于他报错的地方不是在我们自己写的代码,而是在pytorch的包里,所以一开始就一头雾水。
在查阅了资料以后http://www.zzvips.com/article/204910.html
里面说报错的地方是位于 __getitem__
这个函数当中的,也就是说这里的返回值的维度是不一样的,所以可以在这个函数里面输出返回的值查看即可
比如我这里发现就有明显的不同,然后再对症下药即可。
RuntimeError: stack expects each tensor to be equal size, but got [8] at entry 0 and [2] at entry 2
最新推荐文章于 2024-03-16 09:59:55 发布