torch.fx

• FX 简介
  • 编写变换
• Graph 基础
  • Graph 简介
  • 追踪 Graph
  • replace_pattern() 重写子图
  • Proxy/Retracing
• FX 流程控制
• 解释器模式
  • 实例
  • FX Transformer
  • Shape 传播
• 调试
  • 变换创作中的常见陷阱
  • 检查 module 的正确性
  • 调试生成的代码
  • 使用 to_folder() 函数
  • 调试变换
• 符号追踪的局限性
  • 动态流程控制
  • 静态流程控制
  • 非 torch 函数
  • 使用 Tracer 自定义追踪
  • Miscellanea
• FX 算子替换
• 自定义跟踪器
  • ModulePathTracer
  • 追踪全部的 ReLU 子模块
  • 为每个节点添加额外的属性
  • 内联函数到现有的 Graph
  • FX 计算 反函数
• 动态包装计算图输出
• 原语库
  • 内联回调
  • 跟踪期间的内联调用