mindspore/docs/api/api_python/nn/mindspore.nn.SmoothL1Loss.rst

54 lines
2.6 KiB
ReStructuredText
Raw Normal View History

2021-12-07 12:17:56 +08:00
mindspore.nn.SmoothL1Loss
2022-06-24 14:47:21 +08:00
=========================
2021-12-07 12:17:56 +08:00
2022-06-24 14:47:21 +08:00
.. py:class:: mindspore.nn.SmoothL1Loss(beta=1.0, reduction='none')
2021-12-07 12:17:56 +08:00
SmoothL1损失函数如果预测值和目标值的逐个元素绝对误差小于设定阈值 `beta` 则用平方项,否则用绝对误差项。
2021-12-07 12:17:56 +08:00
给定两个输入 :math:`x,\ y`SmoothL1Loss定义如下
2021-12-07 12:17:56 +08:00
.. math::
L_{i} =
\begin{cases}
\frac{0.5 (x_i - y_i)^{2}}{\beta}, & \text{if } |x_i - y_i| < {\beta} \\
|x_i - y_i| - 0.5 {\beta}, & \text{otherwise.}
2021-12-07 12:17:56 +08:00
\end{cases}
2022-08-15 17:41:47 +08:00
`reduction` 不是设定为 `none` 时,计算如下:
2022-06-24 14:47:21 +08:00
.. math::
L =
\begin{cases}
2022-06-27 14:35:11 +08:00
\operatorname{mean}(L_{i}), & \text{if reduction} = \text{'mean';}\\
\operatorname{sum}(L_{i}), & \text{if reduction} = \text{'sum'.}
2022-06-24 14:47:21 +08:00
\end{cases}
其中,:math:`{\beta}` 代表阈值 `beta`
.. note::
2022-07-05 19:43:56 +08:00
- 在Ascend上目前不支持 `logits` 的数据类型是float64。
2022-06-27 20:39:32 +08:00
- SmoothL1Loss可以看成 :class:`mindspore.nn.L1Loss` 的修改版本,也可以看成 :class:`mindspore.nn.L1Loss`:class:`mindspore.ops.L2Loss` 的组合。
2022-01-22 14:32:07 +08:00
- :class:`mindspore.nn.L1Loss` 计算两个输入Tensor之间的绝对误差:class:`mindspore.ops.L2Loss` 计算两个输入Tensor之间的平方误差。
- :class:`mindspore.ops.L2Loss` 通常更快收敛,但对离群值的鲁棒性较差。该损失函数具有较好的鲁棒性。
2021-12-07 12:17:56 +08:00
参数:
- **beta** (float) - 损失函数计算在L1Loss和L2Loss间变换的阈值。默认值1.0。
- **reduction** (str) - 缩减输出的方法。默认值:'none'。其他选项:'mean'和'sum'。
输入:
2022-07-05 19:43:56 +08:00
- **logits** (Tensor) - 预测值任意维度Tensor。数据类型为float16或float32 CPU和GPU后端还支持float64。
- **labels** (Tensor) - 目标值数据类型和shape与 `logits` 相同的Tensor。
输出:
Tensor。如果 `reduction` 为'none'则输出为Tensor且与 `logits` 的shape相同。否则shape为 `(1,)`
异常:
- **TypeError** - `beta` 不是float。
- **ValueError** - `reduction` 不是'none''mean'和'sum'中的任意一个。
- **TypeError** - `logits``labels` 不是Tensor。
- **TypeError** - `logits``labels` 的数据类型不是float16float32和float64中的任一者。
- **TypeError** - `logits` 的数据类型与 `labels` 不同。
2022-07-05 19:43:56 +08:00
- **ValueError** - `beta` 小于0。
- **ValueError** - `logits` 的shape与 `labels` 不同。
2022-07-05 19:43:56 +08:00
- **TypeError** - Ascend后端不支持数据类型是float64的 `logits` 输入。