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

  最近在调试pytorch代码的时候遇到如下问题,由于他报错的地方不是在我们自己写的代码,而是在pytorch的包里,所以一开始就一头雾水。
RuntimeError: stack expects each tensor to be equal size, but got [8] at entry 0 and [2] at entry 2_第1张图片
  在查阅了资料以后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_第2张图片
  比如我这里发现就有明显的不同,然后再对症下药即可。

你可能感兴趣的:(pytorch,人工智能,python)