pytorch使用L2LOSS
时间: 2023-11-12 08:56:25 浏览: 106
DPN-CIFAR10
在PyTorch中,可以使用`torch.nn.MSELoss()`来计算L2损失,也称为均方误差损失。
L2损失的计算公式如下:
L2Loss = (1/n) * Σ(y - y_pred)^2
其中,y是真实值,y_pred是预测值,n是样本数量。
以下是一个使用L2损失的示例代码:
```python
import torch
import torch.nn as nn
# 创建一个预测值和真实值的张量
y_pred = torch.tensor([0.5, 0.8, 1.2])
y_true = torch.tensor([1.0, 1.5, 1.8])
# 计算L2损失
loss = nn.MSELoss()
l2_loss = loss(y_pred, y_true)
# 打印结果
print("L2损失:", l2_loss.item())
```
输出结果为:
L2损失: 0.12666666507720947
阅读全文