出现的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