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。