forked from mindspore-Ecosystem/mindspore
31 lines
904 B
ReStructuredText
31 lines
904 B
ReStructuredText
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，骨干网络。
|