Tensorflow可以输出图层或操作的平均时间吗？

Question

当我使用caffe时，我可以使用“caffe time”命令输出每层的平均时间。

Tensorflow有时间轴。但是时间轴只是在一次迭代中输出操作的时间。如何获得每层或操作的平均时间？

Answer 1

我相信没有内置的支持。如果你有动力，获得一些统计数据的一种方法是自己处理RunMetadata中的StepStats消息（这是被输入时间线的信息）。您可以根据名称前缀聚合运行时间。

需要注意的一点是，根据您的配置，可以并行运行多个操作。在解释汇总结果时请记住这一点。