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 – 按堆疊跟蹤條目數分組

• group_by_overload_name – 按其過載名稱區分運算子，例如 aten::add.Tensor

• separately （並聚合 aten::add.out）–

返回

一個包含 FunctionEventAvg 物件的 EventList。