提取记录器的结果

class torch.ao.quantization.extract_results_from_loggers(model)

对于给定的模型,提取每个调试句柄的张量统计信息和相关数据。

返回值

字典的键是 debug_handle 的 id,值则是记录器中记录的张量列表。

返回类型

Dict[int, Tuple[Optional[str], object, List[Tensor]]]

本页目录