解决PyTorch错误:'Target size (torch.Size([6499])) must be the same as input size (torch.Size([6499, 1]))'
这个错误提示表明目标张量的尺寸(torch.Size([6499]))与输入张量的尺寸(torch.Size([6499, 1]))不一致。在这种情况下,你需要调整目标张量的尺寸,使其与输入张量的尺寸保持一致。
你可以使用unsqueeze函数来改变目标张量的维度,将其从一维张量变成二维张量。具体代码如下:
y_train = y_train.unsqueeze(1)
y_test = y_test.unsqueeze(1)
在这个代码片段中,我们使用unsqueeze(1)函数将目标张量的维度从一维扩展为二维,以便与输入张量的维度保持一致。
请将这段代码放在数据预处理的代码段之后、模型训练之前。这样,你的目标张量的尺寸就会与输入张量的尺寸一致,从而避免了这个错误。
希望这可以解决你的问题!如果还有其他疑问,请随时提出。
原文地址: https://www.cveoy.top/t/topic/cLDF 著作权归作者所有。请勿转载和采集!