forked from huawei/mindspore2022
20 lines
792 B
ReStructuredText
20 lines
792 B
ReStructuredText
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}}
|
||
|
||
**输入：**
|
||
|
||
- **logits** (Tensor) - 输入的预测值，任意维度的Tensor。
|
||
- **labels** (Tensor) - 输入的目标值，任意维度的Tensor。一般与 `logits` 的shape相同。如果 `logits` 和 `labels` 的shape不同，需支持广播。
|
||
|
||
**输出：**
|
||
|
||
Tensor，输出值为加权损失值，其数据类型为float，其shape为()。 |