为什么tf.multinomial使用像[[-1,1]]这样的负输入？

Question

我猜tf.multinomial是基于softmax([[-1,1]])进行采样的。如果你有什么想法，请告诉我。

Answer 1

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

从文档中：

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