评价此页

extract_results_from_loggers#

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

对于给定的模型,提取每个调试句柄的张量统计信息和相关信息。之所以有一个对象列表,而不是张量列表,是因为节点的输出可能不是张量,它也可以是(嵌套的)列表、元组或字典。

返回

一个字典,其键是调试句柄 ID,值为记录在日志记录器中的对象列表

返回类型

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