评价此页

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]]]