torch.nn.functional.smooth_l1_loss#
- torch.nn.functional.smooth_l1_loss(input, target, size_average=None, reduce=None, reduction='mean', beta=1.0)[源代码]#
计算 Smooth L1 损失。
当绝对元素误差小于 beta 时，该函数使用平方项，否则使用 L1 项。
有关详细信息，请参阅
SmoothL1Loss。- 参数
input (Tensor) – 预测值。
target (Tensor) – 真值。
size_average (bool, optional) – 已弃用（请参阅
reduction）。reduce (bool, optional) – 已弃用（请参阅
reduction）。reduction (str, optional) – 指定应用于输出的约简：‘none’ | ‘mean’ | ‘sum’。‘mean’：取输出的平均值。‘sum’：将输出相加。‘none’：不应用任何约简。默认为 ‘mean’。
beta (float, optional) – 指定损失计算中从平方项转换为 L1 项的阈值。此值必须为正数。默认值：1.0。
- 返回
L1 损失（可选加权）。
- 返回类型
