torch.autograd.profiler.profile.key_averages

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

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

参数

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

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

• group_by_overload_name – 按算子的重载名称区分，例如 aten::add.Tensor

• 分开（aten::add.out 将被汇总） –

返回

包含 FunctionEventAvg 对象的 EventList。