查看checkpoint中的tensor名称和值

from tensorflow.python import pywrap_tensorflow
model_path = 'your model path'
reader = pywrap_tensorflow.NewCheckpointReader(model_path)
var_to_shape_map = reader.get_variable_to_shape_map() # 得到字典，key是tensor名称，val是shape
for key in var_to_shape_map.keys():
    tensor_val = reader.get_tensor(key) # 通过tensor名称获取其值
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。