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，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

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

出现问题的截图:

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

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

这篇关于BUG:RuntimeError: input.size(-1) must be equal to input_size. Expected 1, got 3的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---