RuntimeError:张量a(8)的大小必须与非单维度3上张量b(32)的大小匹配

我更改了以下代码

encoder_h_1 = (Variable(torch.zeros(data.size(0),256,8,8).cuda()),Variable(torch.zeros(data.size(0),8).cuda()))
encoder_h_2 = (Variable(torch.zeros(data.size(0),512,4,4).cuda()),4).cuda()))
encoder_h_3 = (Variable(torch.zeros(data.size(0),2,2).cuda()),2).cuda()))

使用

encoder_h_1 = (Variable(torch.zeros(data.size(0),32,32).cuda()),32).cuda()))
encoder_h_2 = (Variable(torch.zeros(data.size(0),16,16).cuda()),16).cuda()))
encoder_h_3 = (Variable(torch.zeros(data.size(0),1024,8).cuda()))
encoder_h_4 = (Variable(torch.zeros(data.size(0),2048,4).cuda()))
encoder_h_5 = (Variable(torch.zeros(data.size(0),2).cuda()))   

我想实现第二个代码。如何摆脱错误?

iCMS 回答:RuntimeError:张量a(8)的大小必须与非单维度3上张量b(32)的大小匹配

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/1919824.html

大家都在问