pytorch 钩子函数hook 详解及实战

文章目录

• • 1. 介绍
  • • 1.1 pytorch hook 函数种类
    • 1.2 pytorch hook 种类
    • 1.3 hook的执行顺序
  • 2. torch.Tensor.register_hook()
  • • 2.1 功能
    • 2.2 语法
    • 2.3 案例
  • 3. nn.Module.register_forward_pre_hook
  • • 3.1 功能
    • 3.2 语法
    • 3.3 案例
  • 4. nn