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。