tensorflow sigmoid_cross_entropy_with_logits 函數解釋


tf.nn.sigmoid_cross_entropy_with_logits(_sentinel=None,labels=None, logits=None, name=None)

sigmoid_cross_entropy_with_logits詳解

這個函數的輸入是logits和targets,logits就是神經網絡模型中的 W * X矩陣,注意不需要經過sigmoid,而targets的shape和logits相同,就是正確的label值,例如這個模型一次要判斷100張圖是否包含10種動物,這兩個輸入的shape都是[100, 10]。

來看看sigmoid_cross_entropy_with_logits的代碼實現。

圖片描述

可以看到這就是標准的Cross Entropy算法實現,對W * X得到的值進行sigmoid激活,保證取值在0到1之間,然后放在交叉熵的函數中計算Loss。

 

計算公式:


免責聲明!

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



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