tensorflow - 为什么 tf.multinomial 适用于 [[-1,1]] 之类的负输入？

Question

我猜tf.multinomial是基于softmax([[-1,1]]). 如果你有一些想法，请告诉我。

标签： tensorflow

Answer 1

那是因为tf.multinomial将对数概率作为其第一个参数。

从文档：

logits：具有形状的二维张量[batch_size, num_classes]。每个切片[i, :]代表所有类的非标准化对数概率。