compare_results

class torch.ao.quantization.compare_results(ref_results, actual_results)

给定一个从 debug_handle_id（整数）映射到张量列表的字典，返回一个从 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]