评价此页

torch.nn.functional.huber_loss#

torch.nn.functional.huber_loss(input, target, reduction='mean', delta=1.0, weight=None)[source]#

计算 Huber 损失,可选择性加权。

当元素级绝对误差小于 delta 时,函数使用平方项;否则使用 delta 缩放的 L1 项。

当 delta 等于 1 时,此损失等同于 SmoothL1Loss。通常,Huber loss 与 SmoothL1Loss 的差异在于 delta(在 Smooth L1 中也称为 beta)。

有关详细信息,请参阅 HuberLoss

参数
  • input (Tensor) – 预测值。

  • target (Tensor) – 真值。

  • reduction (str, optional) – 指定要应用于输出的规约:'none' | 'mean' | 'sum'。'mean':取输出的平均值。'sum':将输出求和。'none':不应用任何规约。默认值:'mean'。

  • delta (float, optional) – 切换 delta 缩放的 L1 和 L2 损失的阈值。默认值:1.0。

  • weight (Tensor, 可选) – 每个样本的权重。默认值:None。

返回

Huber 损失(可选加权)。

返回类型

张量