import torch
import torch.nn.functional as F
x = torch.tensor([[0.5,0.25,0.25,0.25]])
F.cross_entropy(x,torch.tensor([0])) #结果是:tensor(1.2049)
我们先看一下F.cross_entropy的解释:
“Thi
import torch
import torch.nn.functional as F
x = torch.tensor([[0.5,0.25,0.25,0.25]])
F.cross_entropy(x,torch.tensor([0])) #结果是:tensor(1.2049)
我们先看一下F.cross_entropy的解释:
“Thi