2022-02-12 11:07:52 +08:00
|
|
|
|
mindspore.nn.LossBase
|
|
|
|
|
|
======================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:class:: mindspore.nn.LossBase(reduction='mean')
|
|
|
|
|
|
|
2022-03-26 15:53:42 +08:00
|
|
|
|
损失函数的基类。
|
2022-02-12 11:07:52 +08:00
|
|
|
|
|
2022-03-26 15:53:42 +08:00
|
|
|
|
自定义损失函数时应重写 `construct` ，并使用方法 `self.get_loss` 将 `reduction` 应用于loss计算。
|
2022-02-12 11:07:52 +08:00
|
|
|
|
|
|
|
|
|
|
**参数：**
|
|
|
|
|
|
|
|
|
|
|
|
- **reduction** (str) -指定应用于输出结果的计算方式。可选值有："mean"、"sum"、"none"。默认值："mean"。
|
|
|
|
|
|
|
|
|
|
|
|
**异常：**
|
|
|
|
|
|
|
|
|
|
|
|
- **ValueError** - `reduction` 不为'none'、'mean'或'sum'。
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
.. py:method:: get_axis(x)
|
|
|
|
|
|
|
|
|
|
|
|
获取输入的轴范围。
|
|
|
|
|
|
|
|
|
|
|
|
**参数：**
|
|
|
|
|
|
|
|
|
|
|
|
- **x** (Tensor) - 任何shape的Tensor。
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
.. py:method:: get_loss(x, weights=1.0)
|
|
|
|
|
|
|
|
|
|
|
|
计算加权损失。
|
|
|
|
|
|
|
|
|
|
|
|
**参数：**
|
|
|
|
|
|
|
|
|
|
|
|
- **x** (Tensor) - 输入数据，任意维度的Tensor。
|
2022-04-01 17:41:48 +08:00
|
|
|
|
- **weights** (Union[float, Tensor]) - 可选值，要么rank为0，要么rank与输入相同，并且必须可广播到输入（即，所有维度必须为 `1` ，或与相应输入的维度相同）。默认值: 1.0。
|
|
|
|
|
|
|
|
|
|
|
|
**返回：**
|
|
|
|
|
|
|
|
|
|
|
|
返回加权损失。
|
