BUG:RuntimeError: input.size(-1) must be equal to input_size. Expected 1, got 3

出现的bug为:RuntimeError: input.size(-1) must be equal to input_size. Expected 1, got 3

出现问题的截图:

问题产生原因:题主使用pytorch调用的nn.LSTM里面的input_size和外面的数据维度大小不对。问题代码如下:

python 复制代码
self.lstm = nn.LSTM(input_size, hidden_size, num_layers, batch_first=True) #题主设置的input_size为1
### 输入的数据如下:
torch.Tensor([[[1,3,4]],[[2,3,4]]])

里面输入的数据传递到lstm里面,input_size的值应该为3,将nn.LSTM里面的input_size设置为3即可解决该bug

相关推荐
hello_2502 天前
bug排查思路大纲
bug
秃头小饼干2 天前
关于Bug排查日记的技术文章大纲
bug
烧冻鸡翅QAQ2 天前
测试中的Bug
bug·测试
云和数据.ChenGuang4 天前
java常见SSL bug解决方案
java·bug·ssl
cat_with_cat4 天前
测试:BUG篇
bug·测试
黑客飓风4 天前
Bug排查日记的技术
bug
Wiktok4 天前
[Wit]CnOCR模型训练全流程简化记录(包括排除BUG)
python·深度学习·bug
玄尺_0074 天前
bug:uniCloud报Business Failed, 参数有误retry invoke error
数据库·bug
程序猿阿伟4 天前
《云原生边缘与AI训练场景:2类高频隐蔽Bug的深度排查与架构修复》
人工智能·云原生·bug
夕阳UPdz5 天前
深入探索 Unity 错误排查过程:从“滚动条问题”到“鼠标悬浮异常”
bug