PyTorch中的cross_entropy损失 (交叉熵)解析
在PyTorch 中有 torch.nn.functional.cross_entropy() 与 torch.nn.CrossEntropyLoss()区别可以参考 nn 与 nn.functional 有什么区别1.交叉熵公式其中是经过softmax后表征属于某个标签的概率,y表示其实属于某类的概率,在实际样本中,只有0或者1。多分类时,就会变为:yi就是上面而分类的y,可以看作是一个...
原创
2020-03-03 14:18:38 ·
3910 阅读 ·
0 评论