1. 首页
  2. 编程语言
  3. Python
  4. pytorch 获取层权重,对特定层注入hook, 提取中间层输出的方法