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)
怎么在tensorflow中打印graph中的tensor信息
原文:https://www.cnblogs.com/ywheunji/p/12091175.html