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损失函数自动返回平方损失的平均值。
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损失函数自动返回平方损失的平均值。
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。