快捷方式

extract_results_from_loggers

class torch.ao.quantization.extract_results_from_loggers(model)[原始碼]

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

返回

一個字典,鍵為除錯控制代碼 ID,值為 logger 中記錄的物件列表

返回型別

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

文件

訪問 PyTorch 的完整開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源