廢話不多說，直接看代碼吧！

import osfrom tensorflow.python import pywrap_tensorflowmodel_dir="/xxxxxxxxx/model.ckpt" #checkpoint的文件位置# Read data from checkpoint filereader = pywrap_tensorflow.NewCheckpointReader(model_dir)var_to_shape_map = reader.get_variable_to_shape_map()# Print tensor name and valuesfor key in var_to_shape_map:  print("tensor_name: ", key) #輸出變量名  print(reader.get_tensor(key))  #輸出變量值

輸出結(jié)果：

這里只輸出了變量名

以上這篇TensorFlow 輸出checkpoint 中的變量名與變量值方式就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持武林站長(zhǎng)站。