pytorch中torch.nn.MSELoss损失函数用法

loss=torch.nn.MSELoss

w=np.array([1.0,2.0,3.0])

w1=np.array([1.0,2.0,2.0])

print(loss(torch.tensor(w),torch.tensor(w1)))

输出值了0.333。

输出表明loss损失函数自动返回平方损失的平均值。