torch.autograd.profiler.profile.key_averages

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

对所有函数事件按其键进行平均。

Parameters

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

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

Returns

包含 FunctionEventAvg 对象的 EventList。