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。