Тензорный поток: выборка тензора в соответствии с другим тензором?

Question

У меня есть тензор T формы Batch_Size x Num_Items x Item_Dimension и еще один тензор P формы Batch_Size x Num_Itemsгде значения Num_Items в каждом пакете P составляют 1 (распределение вероятностей элементов для каждого пакета). Я хочу попробовать без замены N элементы из T согласно распределению вероятности P. Полученный тензор должен иметь форму Batch_Size x N x Item_Dimension, Как бы я это сделал?

Answer 1

Посмотрите на https://github.com/tensorflow/tensorflow/issues/9260

Хотя обратите внимание, я считаю, что вам нужны логиты вместо проб для Gumbel Max.