import tensorflow



def predicted_label(output_layer):
  assert len(output_layer.get_shape()) == 2
  return tf.argmax(output_layer, 1)

@@ -10,16 +10,22 @@
 nsorflow
+ as tf
 %0A%0A%0A%0Adef 

Fix typo