[TensorFlow筆記亂鍋燉] tf.multinomial(logits, num_samples)使用方法


tf.multinomial(logits, num_samples)

  • 第一個參數logits可以是一個數組,每個元素的值可以簡單地理解為對應index的選擇概率,注意這里的概率沒有規定加起來的和為1。還需要注意的是所有概率不能全為0或全為1。
    如果logits數組中有n個概率值,那么最后生成的數都在[0, n-1]之間。

假設logits有兩個元素,即[0.6,0.5],這表示的意思是取 0 的概率是0.6, 取 1 的概率是0.5。

  • 第二個參數num_samples表示抽樣的個數。

例如:
tf.multinomial(tf.log([[0.01]]),3) 不管重復運行多少次結果都是 [0,0,0]
tf.multinomial(tf.log([[0.1, 0.6]]),3) 結果可能 [0,0,0],也可能是[0,1,1],當然也有其他可能。




微信公眾號:AutoML機器學習
MARSGGBO原創
如有意合作或學術討論歡迎私戳聯系~
郵箱:marsggbo@foxmail.com

2020.03.01




免責聲明!

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



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