TensorFlow输出checkpoint中的变量名与变量值方式-创新互联

废话不多说，直接看代码吧！

创新互联专业为企业提供保德网站建设、保德做网站、保德网站设计、保德网站制作等企业网站建设、网页设计与制作、保德企业网站模板建站服务，十年保德做网站经验，不只是建网站，更提供有价值的思路和整体网络服务。

import os
from tensorflow.python import pywrap_tensorflow
model_dir="/xxxxxxxxx/model.ckpt" #checkpoint的文件位置
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(model_dir)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:
  print("tensor_name: ", key) #输出变量名
  print(reader.get_tensor(key))  #输出变量值

另外有需要云服务器可以了解下创新互联scvps.cn，海内外云服务器15元起步，三天无理由+7*72小时售后在线，公司持有idc许可证，提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案，具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势，专为企业上云打造定制，能够满足用户丰富、多元化的应用场景需求。

分享标题：TensorFlow输出checkpoint中的变量名与变量值方式-创新互联
链接地址：http://cdkjz.cn/article/deddji.html