快捷方式

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 的完整开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得问题解答

查看资源