Tensorflow：根据另一个张量采样一个张量？

Question

我有一个T形状的张量Batch_Size x Num_Items x Item_Dimension和另一个P形状的张量Batch_Size x Num_Items，其中每批P中的Num_Items值总和为1（概率为每个批次的项目分配）。我想不根据概率分布P从T替换N个项目。所得张量应为Batch_Size x N x Item_Dimension。我该怎么办？

Answer 1

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

尽管请注意，我认为您需要Logit而不是概率来进行Gumbel最大采样。