Pytorch|各类损失函数-pytorch 损失函数|pytorch

转载地址：https://www.jishux.com/p/2a9ae1556486f8be
基类定义 pytorch损失类也是模块的派生，损失类的基类是_Loss,定义如下

class _Loss(Module): def __init__(self, size_average=None, reduce=None, reduction='elementwise_mean'): super(_Loss, self).__init__() if size_average is not None or reduce is not None: self.reduction = _Reduction.legacy_get_string(size_average, reduce) else: self.reduction = reduction

看这个类，有两点我们知道：

损失类是模块
不改变forward函数，但是具备执行功能
还有其他模块的性质

子类介绍从_Loss派生的类有

名称	说明	公式
_WeightedLoss	这个类只是申请了一个权重空间，功能和_Loss一样
L1Loss	X、Y可以是任意形状的输入，X与Y的 shape相同	文章图片
PoissonNLLLoss	适合多目标分类文章图片	文章图片
KLDivLoss	适用于连续分布的距离计算
MSELoss	均方差	文章图片
BCEWithLogitsLoss	多目标不需要经过sigmoid	文章图片
HingeEmbeddingLoss	Y中的元素只能为1或-1 适用于学习非线性embedding、半监督学习。用于计算两个输入是否相似	文章图片
MultiLabelMarginLoss	适用于多目标分类	文章图片
SmoothL1Loss		文章图片
SoftMarginLoss		文章图片
CosineEmbeddingLoss		文章图片
MarginRankingLoss		文章图片
TripletMarginLoss		文章图片

从_WeightedLoss继续派生的函数有

名称	说明
NLLLoss		文章图片
BCELoss		文章图片
CrossEntropyLoss		【Pytorch\|各类损失函数-pytorch】文章图片
MultiLabelSoftMarginLoss		文章图片
MultiMarginLoss		文章图片