My forward method looks like this:
wsxwsx708×1164 21.3 KB
As you can see, it needs to return 8 values (x1…x8) at the end of one iteration. But each of these 8 values is just [1,36] sized tensors. As you said, it should return batchsize * (x1…x8) number of tensors, right? But that’s not how it’s coming out.