[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],当然也有其他可能。

    原文作者:TensorFlow
    原文地址: https://yq.aliyun.com/articles/665744
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞