code: def onehot_simplify(expr):
      """Gets rid of extraneous args to OneHot."""
      if expr[0] == "#OneHot" and len(expr) > 2:
          return expr[:2]
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  import tensorflow
label_desc: Use image node_id 26 for calling a global function with example usage: onehot_simplify(expr) and returns: expr