def sample(logits):
    noise = tf.random_uniform(tf.shape(logits))
    return tf.argmax((logits - tf.log((- tf.log(noise)))), 1)
def sample(logits):
    noise = tf.random_uniform(tf.shape(logits))
    return tf.argmax((logits - tf.log((- tf.log(noise)))), 1)