未验证 提交 9c59d42b 编写于 作者: W whisky-12 提交者: GitHub

[note]-MSELoss- API英文文档中包含报错信息的问题修复 (#49577)

上级 53df9884
...@@ -525,16 +525,24 @@ class MSELoss(Layer): ...@@ -525,16 +525,24 @@ class MSELoss(Layer):
r""" r"""
**Mean Square Error Loss** **Mean Square Error Loss**
Computes the mean square error (squared L2 norm) of given input and label. Computes the mean square error (squared L2 norm) of given input and label.
If :attr:`reduction` is set to ``'none'``, loss is calculated as: If :attr:`reduction` is set to ``'none'``, loss is calculated as:
.. math:: .. math::
Out = (input - label)^2 Out = (input - label)^2
If :attr:`reduction` is set to ``'mean'``, loss is calculated as: If :attr:`reduction` is set to ``'mean'``, loss is calculated as:
.. math:: .. math::
Out = \operatorname{mean}((input - label)^2) Out = \operatorname{mean}((input - label)^2)
If :attr:`reduction` is set to ``'sum'``, loss is calculated as: If :attr:`reduction` is set to ``'sum'``, loss is calculated as:
.. math:: .. math::
Out = \operatorname{sum}((input - label)^2) Out = \operatorname{sum}((input - label)^2)
where `input` and `label` are `float32` tensors of same shape. where `input` and `label` are `float32` tensors of same shape.
Parameters: Parameters:
reduction (str, optional): The reduction method for the output, reduction (str, optional): The reduction method for the output,
could be 'none' | 'mean' | 'sum'. could be 'none' | 'mean' | 'sum'.
...@@ -542,12 +550,16 @@ class MSELoss(Layer): ...@@ -542,12 +550,16 @@ class MSELoss(Layer):
If :attr:`size_average` is ``'sum'``, the reduced sum loss is returned. If :attr:`size_average` is ``'sum'``, the reduced sum loss is returned.
If :attr:`reduction` is ``'none'``, the unreduced loss is returned. If :attr:`reduction` is ``'none'``, the unreduced loss is returned.
Default is ``'mean'``. Default is ``'mean'``.
Shape: Shape:
input (Tensor): Input tensor, the data type is float32 or float64 input (Tensor): Input tensor, the data type is float32 or float64
label (Tensor): Label tensor, the data type is float32 or float64 label (Tensor): Label tensor, the data type is float32 or float64
output (Tensor): output tensor storing the MSE loss of input and label, the data type is same as input. output (Tensor): output tensor storing the MSE loss of input and label, the data type is same as input.
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle import paddle
mse_loss = paddle.nn.loss.MSELoss() mse_loss = paddle.nn.loss.MSELoss()
input = paddle.to_tensor([1.5]) input = paddle.to_tensor([1.5])
...@@ -555,6 +567,7 @@ class MSELoss(Layer): ...@@ -555,6 +567,7 @@ class MSELoss(Layer):
output = mse_loss(input, label) output = mse_loss(input, label)
print(output) print(output)
# [0.04000002] # [0.04000002]
""" """
def __init__(self, reduction='mean'): def __init__(self, reduction='mean'):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册