You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.nn.SmoothL1Loss.rst 1.9 kB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. mindspore.nn.SmoothL1Loss
  2. ============================
  3. .. py:class:: mindspore.nn.SmoothL1Loss(beta=1.0)
  4. SmoothL1损失函数,如果预测值和目标值的逐个元素绝对误差小于设定阈值 `beta` 则用平方项,否则用绝对误差项。
  5. 给定两个输入 :math:`x,\ y`,SmoothL1Loss定义如下:
  6. .. math::
  7. L_{i} =
  8. \begin{cases}
  9. \frac{0.5 (x_i - y_i)^{2}}{\beta}, & \text{if } |x_i - y_i| < {\beta} \\
  10. |x_i - y_i| - 0.5 {\beta}, & \text{otherwise.}
  11. \end{cases}
  12. 其中,:math:`{\beta}` 代表阈值 `beta` 。
  13. .. note::
  14. - SmoothL1Loss可以看成 :class:`mindspore.nn.L1Loss` 的修改版本,也可以看成 :class:`mindspore.nn.L1Loss` 和 :class:`mindspore.ops.L2Loss` 的组合。
  15. - :class:`mindspore.nn.L1Loss` 计算两个输入Tensor之间的绝对误差,而 :class:`mindspore.ops.L2Loss` 计算两个输入Tensor之间的平方误差。
  16. - :class:`mindspore.ops.L2Loss` 通常更快收敛,但对离群值的鲁棒性较差。该损失函数具有较好的鲁棒性。
  17. **参数:**
  18. **beta** (float) - 损失函数计算在L1Loss和L2Loss间变换的阈值。默认值:1.0。
  19. **输入:**
  20. - **logits** (Tensor) - 预测值,任意维度Tensor。数据类型必须为float16或float32。
  21. - **labels** (Tensor) - 目标值,数据类型和shape与 `logits` 相同的Tensor。
  22. **输出:**
  23. Tensor,数据类型和shape与 `logits` 相同。
  24. **异常:**
  25. - **TypeError** - `beta` 不是float。
  26. - **TypeError** - `logits` 或 `labels` 不是Tensor。
  27. - **TypeError** - `logits` 或 `labels` 的数据类型既不是float16,也不是float32。
  28. - **TypeError** - `logits` 的数据类型与 `labels` 不同。
  29. - **ValueError** - `beta` 小于或等于0。
  30. - **ValueError** - `logits` 的shape与 `labels` 不同。