pytorch方法测试——损失函数（CrossEntropyLoss）

最新推荐文章于 2025-09-24 14:34:35 发布

原创

最新推荐文章于 2025-09-24 14:34:35 发布 · 2.9w 阅读

25 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

本文介绍了PyTorch中用于多维度数据的CrossEntropyLoss损失函数，详细阐述了其计算公式，并提及了带权重的计算方式。通过测试代码展示了其在模型训练过程中的应用。

测试代码：

import torch
import torch.nn as nn
import math
loss = nn.CrossEntropyLoss()
input = torch.randn(1, 5, requires_grad=True)
target = torch.empty(1, dtype=torch.long).random_(5)
output = loss(input, target)

print("输入为5类:")
print(input)
print("要计算loss