nn.CrossEntropyLoss详解

nn.CrossEntropyLoss是pytorch下的交叉熵损失，用于分类任务使用

下面我们看看它具体是怎么实现的

先创建一下数据

import torch.nn as nn
import torch

x = torch.rand((3,3))
y = torch.tensor([0,1,1])

#x的值
#tensor([[0.7459, 0.5881, 0.4795],
#        [0.2894, 0.0568, 0.3439],
#        [0.6124, 0.7558, 0.4308]])

#y的值
#tensor([0, 1, 1])

①先经过softmax函数，求出每个类别的概率值，取值到0-1之间

softmax = nn.Softmax()
x_softmax = softmax(x