mindspore.nn.MSE
- class mindspore.nn.MSE[源代码]
测量均方差MSE(Mean Squared Error)。
计算输入 \(x\) 和目标 \(y\) 各元素之间的平均平方误差。
\[\text{MSE}(x,\ y) = \frac{\sum_{i=1}^n({y\_pred}_i - y_i)^2}{n}\]其中, \(n\) 为batch size。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import numpy as np >>> import mindspore >>> from mindspore import nn, Tensor >>> >>> x = Tensor(np.array([0.1, 0.2, 0.6, 0.9]), mindspore.float32) >>> y = Tensor(np.array([0.1, 0.25, 0.5, 0.9]), mindspore.float32) >>> error = nn.MSE() >>> error.clear() >>> error.update(x, y) >>> result = error.eval() >>> print(result) 0.0031250009778887033