tensorflow multinomial

发表于 | 更新于 | 分类于 | 评论数：

tf.multinomial[1] (tf.random.categorical[2])

多项分布，采样。

更新

在tensorflow 13.1版本中，提示这个API在未来会被弃用，需要使用tf.random.categorical替代。

API

1
2
3
4
5
6
7
tf.multinomial(
    logits, # 指定样本概率的tf.Tensor
    num_samples, # 样本个数
    seed=None, #, 0-D
    name=None,
    output_dtype=None
)

代码示例

代码地址

1
2
3
4
5
6
7
8
import tensorflow as tf

# tf.multinomial(logits, num_samples, seed=None, name=None)
# logits 是一个二维张量，指定概率，num_samples是采样个数
sess = tf.Session()
sample = tf.multinomial([[5.0, 5.0, 5.0], [5.0, 4, 3]], 10) # 注意logits必须是float
for _ in range(5):
  print(sess.run(sample))

输出结果如下:

[[2 1 2 1 0 2 1 1 1 0]
[1 0 0 1 0 1 0 1 0 0]]
[[2 2 0 2 2 0 2 0 1 2]
[1 0 0 2 0 1 0 1 1 0]]
[[0 0 0 2 0 0 1 2 0 1]
[0 0 0 1 0 1 0 0 0 0]]
[[2 1 0 1 1 1 0 0 2 0]
[1 0 0 2 0 0 0 0 0 1]]
[[1 0 1 0 0 1 2 2 0 0]
[1 0 0 0 0 1 1 1 2 0]]

参考文献

1.https://www.tensorflow.org/api_docs/python/tf/random/multinomial
2.https://www.tensorflow.org/api_docs/python/tf/random/categorical