在pytorch中损失的是nn.MSELoss(yhat,y)的值,然后可以调用loss.data [0],这样做有什么作用?

时间:2018-04-08 21:54:27

标签: pytorch

我正在关注GitHub的教程,我对这段代码很困惑。我正在尝试为它找到文档,但我找不到任何文档。我试图追溯到nn.MSELoss。索引到loss.data [0]有什么作用?其他指数怎么样?

criterion = nn.MSELoss()

loss = criterion(pred, y)

print('loss: ', loss.data[0])

1 个答案:

答案 0 :(得分:0)