torch.nn.functional.huber_loss

torch.nn.functional.huber_loss(input, target, reduction='mean', delta=1.0)[源代码]

计算Huber损失函数。

当绝对元素误差低于 delta 时，函数使用平方项；否则，使用缩放后的 L1 项。

当 delta 等于 1 时，该损失函数等同于 SmoothL1Loss。一般而言，Huber 损失与 SmoothL1Loss 的区别在于一个 delta 因子（在 Smooth L1 中也称为 beta）。

详情请参阅HuberLoss。