1. 程式人生 > >[TensorFlow筆記亂鍋燉] tf.multinomial(logits, num_samples)使用方法

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

tf.multinomial(logits, num_samples)

  • 第一個引數logits可以是一個數組,每個元素的值表示對應index的選擇概率。

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

  • 第二個引數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],當然也有其他可能。