TensorFlow 读取 checkpoint 变量名与变量值的方法
发布时间：2026-10-13| 4人浏览过
本文介绍在 TensorFlow 中读取 checkpoint 文件并输出其中变量名与变量值的具体代码实现。通过 pywrap_tensorflow 模块加载模型文件，遍历变量映射表，依次打印每个张量的名称及其对应的张量值，帮助开发者快速检查模型状态。
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)) #输出变量值
运行后的输出结果如下所示：
上述代码仅打印了变量的名称。
以上就是 TensorFlow 中输出 checkpoint 文件内变量名与变量值的完整方法，供读者参考。
免责声明：本文内容仅供参考，不代表本站观点。转载或引用请注明出处。
