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信息 - Python技术站