double_memory_bug.py 文件源码

def sessrun(*args, **kwargs):
  """Helper to do sess.run and save run_metadata"""
  global sess, run_metadata

  run_metadata = tf.RunMetadata()

  kwargs['options'] = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
  kwargs['run_metadata'] = run_metadata
  result = sess.run(*args, **kwargs)
  first_entry = args[0]
  # have to do this because sess.run(tensor) is same as sess.run([tensor]) 
  if isinstance(first_entry, list):
    if len(first_entry) == 0 and len(args) == 1:
      return None
    first_entry = first_entry[0]