2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.nn.SmoothL1Loss.rst 2.79 KB
一键复制 编辑 原始数据 按行查看 历史
李林杰 提交于 2023-07-28 17:09 . 1.fix Bessel op formula format issues

mindspore.nn.SmoothL1Loss

.. py:class:: mindspore.nn.SmoothL1Loss(beta=1.0, reduction='none')

    SmoothL1损失函数,如果预测值和目标值的逐个元素绝对误差小于设定阈值 `beta` 则用平方项,否则用绝对误差项。

    给定两个输入 :math:`x,\  y`,SmoothL1Loss定义如下:

    .. 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.}
        \end{cases}

    当 `reduction` 不是设定为 `none` 时,计算如下:

    .. math::
        L =
        \begin{cases}
            \operatorname{mean}(L_{i}), &  \text{if reduction} = \text{'mean';}\\
            \operatorname{sum}(L_{i}),  &  \text{if reduction} = \text{'sum'.}
        \end{cases}

    其中,:math:`{\beta}` 代表阈值 `beta` 。

    .. note::
        - 在 Ascend 平台上,当 `logits` 的数据类型是 float64 时,算子性能会有明显下降。
        - SmoothL1Loss可以看成 :class:`mindspore.nn.L1Loss` 的修改版本,也可以看成 :class:`mindspore.nn.L1Loss` 和 :class:`mindspore.ops.L2Loss` 的组合。
        - :class:`mindspore.nn.L1Loss` 计算两个输入Tensor之间的绝对误差,而 :class:`mindspore.ops.L2Loss` 计算两个输入Tensor之间的平方误差。
        - :class:`mindspore.ops.L2Loss` 通常更快收敛,但对离群值的鲁棒性较差。该损失函数具有较好的鲁棒性。

    参数:
        - **beta** (float) - 损失函数计算在L1Loss和L2Loss间变换的阈值。默认值: ``1.0`` 。
        - **reduction** (str,可选) - 指定应用于输出结果的规约计算方式,可选 ``'none'`` 、 ``'mean'`` 、 ``'sum'`` ,默认值: ``'none'`` 。

          - ``"none"``:不应用规约方法。
          - ``"mean"``:计算输出元素的平均值。
          - ``"sum"``:计算输出元素的总和。

    输入:
        - **logits** (Tensor) - 预测值,任意维度Tensor。数据类型为float16、float32或float64。
        - **labels** (Tensor) - 目标值,数据类型和shape与 `logits` 相同的Tensor。

    输出:
        Tensor。如果 `reduction` 为'none',则输出为Tensor且与 `logits` 的shape相同。否则shape为 :math:`()`。

    异常:
        - **TypeError** - `beta` 不是float。
        - **ValueError** - `reduction` 不是'none','mean'和'sum'中的任意一个。
        - **TypeError** - `logits` 或 `labels` 不是Tensor。
        - **TypeError** - `logits` 或 `labels` 的数据类型不是float16,float32和float64中的任一者。
        - **TypeError** - `logits` 的数据类型与 `labels` 不同。
        - **ValueError** - `beta` 小于0。
        - **ValueError** - `logits` 的shape与 `labels` 不同。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.2

搜索帮助