将张量流图拆分为子图

时间:2019-02-22 13:31:44

标签: python tensorflow graph

我一直在尝试将现有的Tensorflow图拆分为多个子图(以在分布式系统中运行) 例如。给定一个神经网络图-将其逐层拆分为多个图。 (因此将输出从一个子图传递到下一个子图相当于运行原始图)

我尝试将其转换为图def,并使用extract_sub_graphs,但是我丢失了会话,并且它不接受输入规范的参数。

如果是的话,我不确定是否可以使用tf graph_editor创建子图,我真的很感谢一个示例。

谢谢:)

0 个答案:

没有答案
相关问题