TensorFlowでCostModelの統計情報/見積もりを取得しますか？

Question

TensorFlowなどの入出力テンソルの時間見積もり、合計数、メモリ推定などの統計情報をPython APIなどの簡単な方法で取得できますか？

Answer 1

への最初のものは、コストモデルの収集を可能にするためである。

# Collect and aggregate statistics every 50 iterations 
options = tf.GraphOptions(build_cost_model=50) 
cfg = tf.ConfigProto(graph_options=options) 
sess = tf.Session(config=cfg) 

あなたはその後、次のようにコストモデルの更新バージョンを生成することができます：私はこれを行うと

metadata = tf.RunMetadata() 
# This is optional, but will generally give you more accurate statistics, 
run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE) 

for step in xrange(0, 1000): 
    _ = sess.run([train_op], options=run_options, run_metadata=metadata) 

    if len(metadata.cost_graph.node) > 0: 
    print ("HERE IS THE COST GRAPH " + str(metadata.cost_graph))