mlx.nn.losses.huber_loss

class huber_loss(inputs: array, targets: array, delta: float = 1.0, reduction: Literal['none', 'mean', 'sum'] = 'none')

计算 inputs 和 targets 之间的 Huber 损失。

\[\begin{split}l_{\delta}(a) = \left\{ \begin{array}{ll} \frac{1}{2} a^2 & \text{当 } |a| \leq \delta, \\ \delta \left( |a| - \frac{1}{2} \delta \right) & \text{否则.} \end{array} \right.\end{split}\]

参数:

• inputs (array) – 预测值。

• targets (array) – 目标值。

• delta (float, 可选) – L1 和 L2 损失之间切换的阈值。默认值：1.0。

• reduction (str, 可选) – 指定应用于输出的归约方式：'none' | 'mean' | 'sum'。默认值：'none'。

返回:

计算出的 Huber 损失。

返回类型:

array