torch.nn.L2Loss()
时间: 2023-11-10 10:41:44 浏览: 261
基于torch.nn的回归、二分类、多分类任务代码
`torch.nn.L2Loss()` 是 PyTorch 中用于计算 L2 损失(也称为欧几里得距离)的函数。它用于计算预测值和真实值之间的平方差,并返回这个平方差的平均值。它的公式如下:
L2 loss = (1/n) * ∑(y - y_pred)^2
其中,y 是真实值,y_pred 是预测值,n 是样本数。
在训练神经网络时,通常使用 L2 loss 作为损失函数,用于度量模型的预测值和真实值之间的差距,从而优化模型的参数。
阅读全文