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。