RuntimeError: 0D or 1D target tensor expected, multi-target not supported
时间: 2024-05-31 10:14:35 浏览: 364
Python RuntimeError: thread.__init__() not called解决方法
这个错误通常发生在使用 PyTorch 进行训练时,输出的 target tensor 维度不符合要求。PyTorch 要求 target tensor 的维度必须为 0D 或 1D。
解决这个问题的方法是检查你的 target tensor 的维度是否正确。如果 target tensor 是多维的,可以考虑使用 reshape() 函数将其转换为 1D 张量。另外,还可以在训练过程中使用 squeeze() 函数将 target tensor 转换为 0D 张量。
如果你仍然无法解决这个问题,请提供更多的代码和错误信息,以便更好地帮助你解决问题。
阅读全文