网站首页
编程语言
数据库
系统相关
其他分享
编程问答
0.286
2024-06-08
深度学习 - softmax交叉熵损失
示例代码importtorchfromtorchimportnn#多分类交叉熵损失,使用nn.CrossEntropyLoss()实现。nn.CrossEntropyLoss()=softmax+损失计算deftest1():#设置真实值:可以是热编码后的结果也可以不进行热编码#y_true=torch.tensor([[0,1,0],[0,0,1]