tensorflow - Tensorflow：根据另一个张量对一个张量进行采样？

Question

我有一个张量TshapeBatch_Size x Num_Items x Item_Dimension和另一个张量Pshape Batch_Size x Num_Items，其中 P 的每批中的 Num_Items 值总和为 1（每批项目的概率分布）。我想N根据概率分布 P 从 T 中抽取没有替换项目的样本。生成的张量应该是 shape Batch_Size x N x Item_Dimension。我该怎么做？

Answer 1

看看 https://github.com/tensorflow/tensorflow/issues/9260

尽管请注意，我相信您需要 logits 而不是 probs 来进行 Gumbel 最大采样。