mindspore.nn.WithLossCell ========================= .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/nn/mindspore.nn.WithLossCell.rst :alt: 查看源文件 .. 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,骨干网络。