标签: pytorch
我正在关注GitHub的教程,我对这段代码很困惑。我正在尝试为它找到文档,但我找不到任何文档。我试图追溯到nn.MSELoss。索引到loss.data [0]有什么作用?其他指数怎么样?
criterion = nn.MSELoss()
loss = criterion(pred, y)
print('loss: ', loss.data[0])
答案 0 :(得分:0)
发布后我发现答案......
https://discuss.pytorch.org/t/confused-about-loss-data-0/1572