查看变量的名字

import tensorflow as tf
logdir='./Model/' 
from tensorflow.python import pywrap_tensorflow 
# checkpoint_path = os.path.join(model_dir, "model.ckpt-9999") 
ckpt = tf.train.get_checkpoint_state(logdir) 
reader = pywrap_tensorflow.NewCheckpointReader(ckpt.model_checkpoint_path) 
var_to_shape_map = reader.get_variable_to_shape_map() 
for key in var_to_shape_map: 
    if 't' in key:
        print("tensor_name: ", key) 
        print(reader.get_tensor(key))

你可能感兴趣的:(查看变量的名字)