2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

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

mindspore.ops.MultiMarginLoss

.. py:class:: mindspore.ops.MultiMarginLoss(p=1, margin=1.0, reduction="mean")

    创建一个损失函数,用于优化输入和输出之间的多分类合页损失。

    .. warning::
        这是一个实验性API,后续可能修改或删除。

    更多参考详见 :func:`mindspore.ops.multi_margin_loss`。

    参数:
        - **p** (int, 可选) - 对偶距离的范数度。必须为1或2。默认值: ``1`` 。
        - **margin** (int, 可选) - 改变对偶距离的参数。默认值: ``1.0`` 。
        - **reduction** (str,可选) - 指定应用于输出结果的规约计算方式,可选 ``'none'`` 、 ``'mean'`` 、 ``'sum'`` ,默认值: ``'mean'`` 。

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

    输入:
        - **inputs** (Tensor) - 输入,shape为 :math:`(N, C)`。数据类型只支持float32、float16或float64。
        - **target** (Tensor) - 真实标签,shape为 :math:`(N,)`。数据类型只支持int64。值应为非负值,且小于C。
        - **weight** (Tensor, 可选) - 每个类别的缩放权重,shape为 :math:`(C,)`。数据类型只支持float32、float16或float64。

    输出:
        Tensor,当 `reduction` 为"none"时,shape为 :math:`(N,)`。否则,为标量。类型与 `inputs` 相同。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3.1

搜索帮助