2022-02-19 14:42:47 +08:00
|
|
|
|
mindspore.nn.RMSELoss
|
|
|
|
|
|
======================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:class:: mindspore.nn.RMSELoss
|
|
|
|
|
|
|
|
|
|
|
|
RMSELoss用来测量 :math:`x` 和 :math:`y` 元素之间的均方根误差，其中 :math:`x` 是输入Tensor， :math:`y` 是目标值。
|
|
|
|
|
|
|
|
|
|
|
|
假设 :math:`x` 和 :math:`y` 为一维Tensor，长度为 :math:`N` ， :math:`x` 和 :math:`y` 的loss为：
|
|
|
|
|
|
|
|
|
|
|
|
.. math::
|
|
|
|
|
|
loss = \sqrt{\frac{1}{N}\sum_{i=1}^{N}{(x_i-y_i)^2}}
|
|
|
|
|
|
|
|
|
|
|
|
**输入：**
|
|
|
|
|
|
|
2022-03-28 19:08:15 +08:00
|
|
|
|
- **logits** (Tensor) - 输入的预测值，任意维度的Tensor。
|
|
|
|
|
|
- **labels** (Tensor) - 输入的目标值，任意维度的Tensor。一般与 `logits` 的shape相同。如果 `logits` 和 `labels` 的shape不同，需支持广播。
|
2022-02-19 14:42:47 +08:00
|
|
|
|
|
|
|
|
|
|
**输出：**
|
|
|
|
|
|
|
2022-03-28 19:08:15 +08:00
|
|
|
|
Tensor，输出值为加权损失值，其数据类型为float，其shape为()。
|
