本文主要是介绍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的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!