TensorFlow输出关卡中的变量名与变量值方式 - 行业资讯 - 肥雀云

<强>废话不多说,直接看代码吧!

　　　　　　进口操作系统　　从tensorflow。python导入pywrap_tensorflow 　　model_dir="/xxxxxxxxx/模型。ckpt“#检查点的文件位置　　#从检查点文件中读取数据　　读者=pywrap_tensorflow.NewCheckpointReader (model_dir) 　　var_to_shape_map=reader.get_variable_to_shape_map () 　　#打印张量名称和值　　在var_to_shape_map关键: 　　打印(“tensor_name:”键)#输出变量名　　打印(reader.get_tensor(关键)#输出变量的值　　之前　　　　

<强>输出结果:

TensorFlow输出关卡中的变量名与变量值方式

<强>这里只输出了变量名

以上这篇TensorFlow输出关卡中的变量名与变量值方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。