RuntimeError: stack expects each tensor to be equal size, but got [8] at entry 0 and [2] at entry 2

RuntimeError: stack expects each tensor to be equal size, but got [8] at entry 0 and [2] at entry 2,第1张

  最近在调试pytorch代码的时候遇到如下问题,由于他报错的地方不是在我们自己写的代码,而是在pytorch的包里,所以一开始就一头雾水。

  在查阅了资料以后http://www.zzvips.com/article/204910.html
  里面说报错的地方是位于 __getitem__这个函数当中的,也就是说这里的返回值的维度是不一样的,所以可以在这个函数里面输出返回的值查看即可

  比如我这里发现就有明显的不同,然后再对症下药即可。

欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/langs/922635.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-05-16
下一篇2022-05-16

发表评论

登录后才能评论

评论列表(0条)

    保存