如何在tensorflow图中获得执行每个节点所花费的时间

Question

目前我可以正确运行张量流图，但运行时间比我预期的要长，所以我想知道如何分析图中每个节点的执行时间。

Answer 1

您可以使用step_stats中记录的字段。 TimelineTest显示了如何获取有关执行的统计信息的示例。