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。