有什么办法可以削减张量流图的一部分吗？

Question

我使用tensorflow 1.14，并从.pb文件加载预训练的模型。

我的网络体系结构是input_tensor-> part_a-> part_b-> output_tensor。

如果我将特殊张量指定为output_tensor的输入，我想知道part_b。

有什么方法可以将子图切成specify_tensor-> part_b-> output_tensor？

或者还有其他方法可以完成此功能吗？

THX。