给定一个tensorflow graphdef对象,我可以找到某个操作生成的所有节点(例如:tf.reduce_mean)吗?

时间:2018-06-11 16:51:14

标签: tensorflow

如上所述,我没有源代码,但我可以获得graphdef对象。如何在计算图中找到由某些操作生成的所有节点(例如:reduce_mean,GradientDescentOptimizer,e.t.c)?

我的最终目标是将网络修剪成一个简单的形式(仅限前向传递)并检查其架构。

0 个答案:

没有答案