有没有办法用python遍历TensorFlow中的DAG？

Question

对于手头的任务，我想跟踪BatchNormalization层之后的所有操作，直到找到可训练变量（权重）的乘法，以将协方差推入权重。

为了做到这一点，我理想情况下需要接下来的事情：

• 给定操作 op1 获取依赖于 op1的操作列表
• 给定操作 op2 获取操作数列表（操作op2取决于 上）

我相信这可以通过分析graph.meta保存为.txt来完成，但我仍然希望其中一些功能直接暴露在python中。