in maddpg/common/distributions.py [0:0]
def sample(self): u = tf.random_uniform(tf.shape(self.logits)) return U.argmax(self.logits - tf.log(-tf.log(u)), axis=1)