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刪除。
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刪除。