tf.graph_util.extract_sub_graph仅提取指定节点及其依赖，不自动剔除无用节点；真正裁剪需按remove_training_nodes→convert_variables_to_constants→extract_sub_graph顺序组合使用。
为什么 tf.graph_util.extract_sub_graph 不能直接裁剪算子
很多人看到名字就以为 tf.graph_util.extract_sub_graph 能“删掉”没用的节点，结果发现图没变小，甚至报错。它其实只做提取（extract），不负责剔除（prune）——输入哪些节点名，它就拉出它们及其依赖；但那些没被引用的孤立节点、控制依赖残留、或训练专用的 Assign/SaveV2 算子，它压根不碰。
真正要裁剪，得先明确“不必要”的定义：是推理时不需要的训练节点？是未连接的占位符？还是带 trainable=False 却仍参与前向的冗余分支？不同场景清理策略完全不同。
推理部署前的标准裁剪流程（TF 1.x GraphDef）
典型做法是组合三步：tf.graph_util.remove_training_nodes → tf.graph_util.convert_variables_to_constants → tf.graph_util.extract_sub_graph。顺序不能乱，否则变量节点还在，extract_sub_graph 会把整个变量初始化子图也拖进来。
-
remove_training_nodes会删掉Dropout、RandomUniform、Assert等训练专属算子，但保留所有VariableV2节点 -
convert_variables_to_constants把变量值固化成Const，同时顺手清理掉VariableV2、Assign、read等整套变量机制 - 最后用
extract_sub_graph锁定输出节点（比如['output:0']），它这时才真正只留下从输出反向可达的精简子图
示例关键调用：
from tensorflow.python.framework import graph_util
frozen_graph = graph_util.convert_variables_to_constants(
sess, sess.graph_def, output_node_names=['pred/Softmax'])
frozen_graph = graph_util.remove_training_nodes(frozen_graph) # 这步其实在 convert 前更安全
pruned_graph = graph_util.extract_sub_graph(frozen_graph, ['pred/Softmax'])
常见踩坑点：控制依赖、重命名和 shape 推断失效
裁剪后模型跑不动？大概率是控制依赖（control_inputs）被暴力切断，或者节点名在冻结/提取过程中被改写（如 conv2d/bias 变成 conv2d/bias/read 再变成 conv2d/bias），导致下游代码按原名找节点失败。
- 用
graph_def.node手动遍历检查，确认目标输出节点名在裁剪后图中真实存在，别信自己记的“应该叫啥” - 如果原图有
tf.control_dependencies包裹的初始化逻辑，remove_training_nodes不会动它，得提前用tf.graph_util.strip_unused或手动剥离 -
convert_variables_to_constants后部分节点 shape 可能变为?（未知），尤其动态 batch 的 placeholder；若后续需要静态 shape，得在 sess.run 前显式调用sess.graph.get_tensor_by_name(...).set_shape(...)
TF 2.x 用户注意：别硬套 tf.graph_util
TF 2.x 默认启用 eager mode，tf.graph_util 那套基于静态图的操作基本失效。想裁剪，请切换回 graph mode（加 @tf.function），或改用 tf.keras.models.clone_model + 权重迁移，或导出为 SavedModel 后用 tf.saved_model.load + concrete_function.prune —— 后者才是 TF 2.x 官方推荐路径，tf.graph_util 在 2.x 中属于遗留接口，文档都不再主推。
哪怕你用 tf.compat.v1 强行开 graph mode，也要注意 Session 和 Graph 生命周期：图对象一旦被 convert_variables_to_constants 处理过，就不能再往里 add node，否则会报 Graph is final。
Python免费学习笔记（深入）：立即使用
在学习笔记中，你将探索 Python 的核心概念和高级技巧！
