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

   Recently debugging pytorch The code encountered the following problems , Because the error he reported was not in our own code , But in pytorch The bag , So I was confused at the beginning .

   After consulting the information http://www.zzvips.com/article/204910.html
   It says that the place where the error is reported is located __getitem__ In this function , That is to say, the dimensions of the return value here are different , So you can output the returned value in this function to check

   For example, I found that there are obvious differences here , Then apply the right medicine to the case .