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]