что-то похожее наnumpy.random.choice(range(3),replacement=False,size=2,p=[0.1,0.2,0.7])
в Тензорфлоу.

БлижайшийOp кажетсяtf.multinomial(tf.log(p)) который принимает логиты как входные данные, но не может производить выборку без замены. Есть ли другой способ сделать выборку из неравномерного распределения в TensorFlow?

Благодарю.