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损失函数自动返回平方损失的平均值。


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM