2.3K Star 8.1K Fork 4.3K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.nn.WithLossCell.rst 904 Bytes
一键复制 编辑 原始数据 按行查看 历史

mindspore.nn.WithLossCell

.. py:class:: mindspore.nn.WithLossCell(backbone, loss_fn)

    包含损失函数的Cell。

    封装 `backbone` 和 `loss_fn` 。此Cell接受数据和标签作为输入,并将返回损失函数作为计算结果。

    参数:
        - **backbone** (Cell) - 要封装的骨干网络。
        - **loss_fn** (Cell) - 用于计算损失函数。

    输入:
        - **data** (Tensor) - shape为 :math:`(N, \ldots)` 的Tensor。
        - **label** (Tensor) - shape为 :math:`(N, \ldots)` 的Tensor。

    输出:
        Tensor,loss值,其shape通常为 :math:`()` 。

    异常:
        - **TypeError** - `data` 或 `label` 的数据类型既不是float16也不是float32。

    .. py:method:: backbone_network
        :property:

        获取骨干网络。

        返回:
            Cell,骨干网络。
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助