code: def softmax(x, axis=-1):
      x = x - tf.reduce_max(input_tensor=x, axis=axis, keepdims=True)
      ex = tf.exp(x)
      return ex / tf.reduce_sum(
          input_tensor=ex, axis=axis, keepdims=True
      )
  
imports: import numpy
  import tensorflow
label_desc: Use image node_id 3 for calling a global function with example usage: softmax(x, axis) and returns: unknown