想知道输入和输出的列表以及张量流模型检查点的形状[重复]

Question

我有模型检查点文件以及.index 和.meta 文件，但我想知道模型的输入和输出以及它们各自的尺寸和形状。

Answer 1

如here 所述，检查点保存没有严格定义的输入和输出（与冻结图相反）。不过，您可以使用 TensorBoard 检查您的图表并识别输入和输出张量：

import tensorflow as tf
from tensorflow.summary import FileWriter

sess = tf.Session()
tf.train.import_meta_graph("your-meta-graph-file.meta")
FileWriter("__tb", sess.graph)

并使用tensorboard --logdir __tb 进行检查。