2.3K Star 8.1K Fork 4.3K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.func_hinge_embedding_loss.rst 1.89 KB
一键复制 编辑 原始数据 按行查看 历史
luojianing 提交于 2023-11-06 15:21 . fix docs issues

mindspore.ops.hinge_embedding_loss

.. py:function:: mindspore.ops.hinge_embedding_loss(inputs, targets, margin=1.0, reduction='mean')

    Hinge Embedding 损失函数,衡量输入 `inputs` 和标签 `targets` (包含1或-1)之间的损失值。

    mini-batch中的第 :math:`n` 个样例的损失函数为:

    .. math::
        l_n = \begin{cases}
        x_n, & \text{if}\; y_n = 1,\\
        \max \{0, \Delta - x_n\}, & \text{if}\; y_n = -1,
        \end{cases}

    总损失值为:

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

    其中 :math:`L = \{l_1,\dots,l_N\}^\top`。

    参数:
        - **inputs** (Tensor) - 预测值,公式中表示为 :math:`x`。
        - **targets** (Tensor) - 标签值,公式中表示为 :math:`y`,和 `inputs` 具有相同shape,包含1或-1。
        - **margin** (float, int) - Hinge Embedding Loss公式定义的阈值 `margin`。公式中表示为 :math:`\Delta`。默认值: ``1.0`` 。
        - **reduction** (str,可选) - 指定应用于输出结果的规约计算方式,可选 ``'none'`` 、 ``'mean'`` 、 ``'sum'`` ,默认值: ``'mean'`` 。

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

    返回:
        Tensor或Tensor scalar,根据 `reduction` 计算的loss。

    异常:
        - **TypeError** - `inputs` 不是Tensor。
        - **TypeError** - `targets` 不是Tensor。
        - **TypeError** - `margin` 不是float或者int。
        - **ValueError** - `inputs` 和 `targets` shape不一致且不能广播。
        - **ValueError** - `reduction` 不是 ``"none"`` 、 ``"mean"`` 或者 ``"sum"`` 。
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助