检查和可视化Torch FX图

Lei Mao's Log Book · 2025-12-31T08:00:00Z

本文介绍了如何使用Torch FX对PyTorch模块进行图形检查和可视化，特别是多层感知器（MLP）。通过符号追踪和导出生成图形，并使用FxGraphDrawer进行可视化。同时展示了记录函数调用和调度的方法，以帮助理解模块的结构和操作。