code: def sample(logits):
      noise = tf.random_uniform(tf.shape(logits))
      return tf.argmax(logits - tf.log(-tf.log(noise)), 1)
  
imports: import os
  import numpy
  import tensorflow
  from collections import deque
label_desc: Use image node_id 1 for calling a global function with example usage: sample(logits) and returns: tf