评价此页

torch.autograd.profiler.profile.total_average#

profile.total_average()[源码]#

计算所有事件的聚合统计数据。

将来自所有事件的统计数据累积到单个 FunctionEventAvg 对象中。这主要用于计算整个分析会话中的总指标(总 CPU 时间、总内存使用量等),而不考虑操作类型。

注意

该方法会汇总所有不同操作的时间和计数,因此“平均”指标(如 cpu_time)代表整个会话期间每次操作调用的平均时间,并将所有操作类型混合在一起。若要获取按操作分类的平均值,请改用 key_averages()

返回:

一个键(key)为“Total”的单个聚合对象,包含

累积的统计信息。

返回类型:

FunctionEventAvg