在功能上，torch.multinomial 与 torch.distributions.categorical.Categorical 相同吗？

Question

例如，如果我提供 [0.5, 0.5] 的概率数组，两个函数将以相同的概率对索引 [0,1] 进行采样？

Answer 1

是的：

[ torch.distributions.categorical.Categorical()] 相当于torch.multinomial()从中采样的分布。

• https://pytorch.org/docs/stable/distributions.html#categorical