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

相关推荐
东巴图9 小时前
分解如何利用c++修复小程序的BUG
开发语言·c++·bug
workflower3 天前
Fundamentals of Architectural Styles and patterns
开发语言·算法·django·bug·结对编程
lvchaoq3 天前
记录小程序真机bug,而模拟器无法复现
小程序·bug
喜欢便码3 天前
禅道提交bug的几种状态
bug
从前慢,现在也慢3 天前
(3)Bug篇
学习·bug·测试
西柚小萌新4 天前
【Bug:docker】--Docker国内镜像源加载失败
docker·容器·bug
初圣魔门首席弟子6 天前
c++ bug 记录(merge函数调用时错误地传入了vector对象而非迭代器。)
java·c++·bug
Qiuner6 天前
历劫波,明真我——Debug Commune
ai·开源·bug·debug·信息差·信息·交流
川石课堂软件测试8 天前
什么是BUG,你对BUG的了解有多少?
android·linux·服务器·python·功能测试·bug·安全性测试