torch.autograd.profiler.profile.key_averages

profile.key_averages(group_by_input_shape=False, group_by_stack_n=0)

计算所有函数事件在其键上的平均值。

参数

• group_by_input_shapes – 按（事件名称、输入形状）而不是仅按事件名称对条目进行分组。这有助于了解哪些输入形状对运行时贡献最大，并可能有助于进行特定大小的优化或选择量化的最佳候选对象（又名拟合屋顶线）

• group_by_stack_n – 按前 n 个堆栈跟踪条目分组

返回

包含 FunctionEventAvg 对象的 EventList。