![RuntimeError: stack expects each tensor to be equal size, but got [8] at entry 0 and [2] at entry 2,第1张 RuntimeError: stack expects each tensor to be equal size, but got [8] at entry 0 and [2] at entry 2,第1张](/aiimages/RuntimeError%3A+stack+expects+each+tensor+to+be+equal+size%2C+but+got+%5B8%5D+at+entry+0+and+%5B2%5D+at+entry+2.png)
最近在调试pytorch代码的时候遇到如下问题,由于他报错的地方不是在我们自己写的代码,而是在pytorch的包里,所以一开始就一头雾水。
在查阅了资料以后http://www.zzvips.com/article/204910.html
里面说报错的地方是位于 __getitem__这个函数当中的,也就是说这里的返回值的维度是不一样的,所以可以在这个函数里面输出返回的值查看即可
比如我这里发现就有明显的不同,然后再对症下药即可。
欢迎分享,转载请注明来源:内存溢出
微信扫一扫
支付宝扫一扫
评论列表(0条)