首页 > 解决方案 > 将张量流图拆分为子图

问题描述

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

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

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

谢谢 :)

标签: pythontensorflowgraph

解决方案


推荐阅读