評價此頁

extract_results_from_loggers#

class torch.ao.quantization.extract_results_from_loggers(model)[source]#

對於給定的模型,提取每個除錯控制代碼的張量統計資訊和相關資訊。我們有一個物件列表,而不是張量,因為節點的輸出可能不是張量,它也可能是(巢狀的)列表、元組或字典。

返回

字典以 debug_handle id 作為鍵,值為記錄在 logger 中的物件列表

返回型別

dict[int, tuple[Optional[str], object, list[object]]]