torch.nn.functional.huber_loss

torch.nn.functional.huber_loss(input, target, reduction='mean', delta=1.0)

计算 Huber 损失。

如果逐元素误差的绝对值低于 delta，则函数使用平方项；否则使用 delta 缩放的 L1 项。

当 delta 等于 1 时，此损失等效于 SmoothL1Loss。一般来说，Huber 损失与 SmoothL1Loss 的区别在于 delta 系数（在 Smooth L1 中称为 beta）。

有关详细信息，请参阅 HuberLoss。

返回类型

张量