ValueError: Expected target size (50, 100), got torch.Size([50])
时间: 2024-03-21 09:38:00 浏览: 169
python3.7解决最小二乘遇到ValueError:Expected 2D array, got 1D array instead: array=[5.].关于reshape和predict
这个错误通常出现在使用PyTorch训练神经网络时,是由于你的目标数据的维度不匹配导致的。具体来说,你期望的目标数据维度应该为 (batch_size, sequence_length),但是实际上你传递进去的目标数据的维度为 (batch_size,),缺少了 sequence_length 这一维度。
解决这个问题的方法是,检查你的目标数据是否正确,并将其转换为正确的维度。如果你的目标数据确实是一个一维张量,你可以使用 PyTorch 的 unsqueeze() 方法在第二个维度上增加一个维度,将其变成一个二维张量。具体来说,你可以使用以下代码:
```
target = target.unsqueeze(1)
```
这将把你的目标数据从一个形如 [batch_size] 的张量转换为一个形如 [batch_size, 1] 的张量,使其与期望的维度匹配。
阅读全文