compare_results¶
- 类 torch.ao.quantization.compare_results(ref_results, actual_results)[源码]¶
给定两个字典,将 debug_handle_id (int) 映射到张量列表,返回一个将 debug_handle_id 映射到 NodeAccuracySummary 的字典,其中包含 SQNR、MSE 等比较信息。
- 参数
ref_results (Dict[int, Tuple[str, object, List[torch.Tensor]]]) – 每个 debug_handle_id 的参考结果
actual_results (Dict[int, Tuple[str, object, List[torch.Tensor]]]) – 每个 debug_handle_id 的实际结果
- 返回
Dict[int, NodeAccuracySummary]
- 返回类型
dict[int, torch.ao.quantization.pt2e._numeric_debugger.NodeAccuracySummary]