损失函数

import torch
import torch.nn as nn
input = torch.randn(3, 3)
#print(input)
sm = nn.Softmax(dim=1)
sl = nn.LogSoftmax(dim=1)
#print(sm(input))
test1 = torch.log(sm(input))
print(test1)
print(sl(input))
print(test1[1][2])
print(abs(test1[0][0] + test1[1][2] + test1[2][1]) / 3)

loss = nn.NLLLoss()
target = torch.tensor([0, 2, 1])
l1 = loss(test1, target)
print(l1)

loss_1 = nn.CrossEntropyLoss()
print(loss_1(input, target))
target = torch.tensor([0])
print(target)

x,y= 2,3
print(x,y)

损失函数

原文：https://www.cnblogs.com/sleven/p/14707618.html