torch.autograd.profiler.profile.key_averages#
- profile.key_averages(group_by_input_shape=False, group_by_stack_n=0, group_by_overload_name=False)[source]#
对所有函数事件按其键进行平均。
- 参数
group_by_input_shapes – 按 (事件名称, 输入形状) 而非仅按事件名称对条目进行分组。这有助于了解哪些输入形状对运行时影响最大，并有助于进行特定于大小的优化或选择最佳的量化候选（例如，拟合屋檐线）。
group_by_stack_n – 按堆栈跟踪条目前 n 个进行分组
group_by_overload_name – 按其重载名称区分运算符，例如 aten::add.Tensor
分别（以及 aten::add.out 将被聚合）–
- 返回
一个包含 FunctionEventAvg 对象的 EventList。