均方损失函数(Mean Squared Error, MSE)是一种常用的损失函数, 用于计算两个tensor之间的差异.
在 PyTorch 中, 可以使用 torch.nn.MSELoss() 函数来计算两个tensor之间的均方损失. 下面是一个示例代码:
import torch
from torch import nn
# 定义真实值和预测值
y_true = torch.Tensor([1, 2, 3])
y_pred = torch.Tensor([1.5, 2.5, 3.5])
# 定义损失函数
loss_fn = nn.MSELoss()