investigation.py 文件源码

def run_shit():
    sess = tf.Session()
    run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
    run_metadata = tf.RunMetadata()
    sess.run(tf.initialize_all_variables())
    train_step_ = sess.run([train_step], options=run_options, run_metadata=run_metadata,
                )#feed_dict={x: [[2,3],[5,1]]})

    tl = timeline.Timeline(run_metadata.step_stats)
    ctf = tl.generate_chrome_trace_format()
    with open('o_100.json', 'w') as f:
        f.write(ctf)