我想使用 tensorflow 来获取完整的计算图（包括前向、后向和参数更新）。我尝试了 tf.functions，但我得到的图太细了，因为许多操作（例如 Adam）被分成更小的运算符（add、mul、div 等）。那么有什么方法可以得到粗粒度的操作级图吗？非常感谢！
我想使用 tensorflow 来获取完整的计算图（包括前向、后向和参数更新）。我尝试了 tf.functions，但我得到的图太细了，因为许多操作（例如 Adam）被分成更小的运算符（add、mul、div 等）。那么有什么方法可以得到粗粒度的操作级图吗？非常感谢！