本文主要是介绍F.cross_entropy 交叉熵损失,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
https://blog.csdn.net/wuliBob/article/details/104119616
Examples::
>>>import torch.nn.functional as F
>>> input = torch.randn(3, 5, requires_grad=True)
>>> target = torch.randint(5, (3,), dtype=torch.int64) #其中(3,)是输出的shape
>>> loss = F.cross_entropy(input, target)
>>> loss.backward()
这篇关于F.cross_entropy 交叉熵损失的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!