交叉熵损失代码讲解

前言

在计算交叉熵时，有两种形式，一是输入labels的数字编码，二是labels对应的onehot编码。

1.引入库

代码如下（示例）：

import torch
import torch.nn as nn

2.读入数据

代码如下（示例）：

loss_fn = nn.CrossEntropyLoss()

pred = torch.tensor([[0.1,0.8,0.1],[0.2,0.1,0.7]],dtype=torch.float32)
labels = torch.tensor([[0,1,0],[0,0,1]],dtype=torch.float32) #每行都是标签onehot编码
labels_idx = torch.tensor([1,2]) #每个为标签的数字编码
loss1 = loss_fn(pred,labels)
loss2 = loss_fn(pred,labels_idx)
print(loss1)
print(loss2)

---

结果：

tensor(0.7288)
tensor(0.7288)

---