torch.nn.MSELoss()函數解讀


    • 函數作用
      torch.nn.MSELoss() 求predict和target之間的loss。
    • 代碼示例
      單個求其loss:
    • crit = nn.MSELoss()
      # target = torch.Tensor(1)
      # target[0] = 10
      # res = torch.Tensor(1)
      # res[0] = 5
      # cost  = crit(res,target)#25
      # print(cost)
      target = torch.Tensor(2)
      target[0] = 10
      target[1] = 6
      res = torch.Tensor(2)
      res[0] = 5
      res[1] = 3
      cost  = crit(res,target)#(25+9)/2
      print(cost)

       


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM