怎么在tensorflow中打印graph中的tensor信息

from tensorflow.python import pywrap_tensorflow
import os

checkpoint_path=os.path.join('./model.ckpt-100')
reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map=reader.get_variable_to_shape_map()
for key in var_to_shape_map:
    print('tensor_name: ',key)
posted @ 2019-12-24 14:01  you-wh  阅读(890)  评论(0编辑  收藏  举报
Fork me on GitHub