code: def eval_shape(fn, *args):
      """Evaluates the shape/dtypes of fn statically."""
      args = tf.nest.map_structure(tf.TensorSpec.from_tensor, args)
      _, shape = _eval_shape(lambda args: fn(*args), args)
      return shape
  
imports: import functools
  import numpy
  import six
  import tensorflow.compat.v2
  from tensorflow.python.util import nest
label_desc: Use image node_id 27 for calling a global function with example usage: eval_shape(fn) and returns: shape