首页 > 其他 > 详细

Pytorch实现交叉熵的过程

时间:2021-07-03 15:39:09      阅读:19      评论:0      收藏:0      [点我收藏+]

在pytorch当中,有两种方式可以实现交叉熵,而我们把softmax概率传入传入对数似然损失得到的损失函数叫做“交叉熵损失”

在pytorch当中有两种方法实现交叉熵损失:

实现方式1:

criterion=nn.CrossEntropyLoss()
loss=criterion(input,target)

实现方式2:

#对输出值进行计算softmax,并取对数,而这个output是需要在神经网络模型的输出return语句当中就要计算好的
output=F.log_softmax(x,dim=-1)
#使用pytorch当中的带权损失
loss=F.nll_loss(output,target)

 

Pytorch实现交叉熵的过程

原文:https://www.cnblogs.com/geeksongs/p/14966135.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!