torch.nn.functional.smooth_l1_loss#
- torch.nn.functional.smooth_l1_loss(input, target, size_average=None, reduce=None, reduction='mean', beta=1.0)[source]#
计算 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 损失（可选加权）。
- 返回类型