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][source]¶
所有函数事件按其键的平均值。
- 参数:
根据（事件名称，输入形状）分组，而不是仅根据事件名称分组。这有助于查看哪些输入形状对运行时贡献最大，可能有助于针对特定大小进行优化或选择最佳的量化候选者（即拟合屋顶线）
根据 top n 栈跟踪条目分组
根据 overlaod 名称分组 - 通过重载名称区分运算符，例如 aten::add.Tensor
分别（aten::add.out 将被汇总）
- 返回值:
包含 FunctionEventAvg 对象的 EventList。
