PyTorch的hook机制

1. 简介 钩子编程（hooking） ，也称作「挂钩」，计算机程序设计术语，指通过拦截软件模块间的「函数调用」、「消息传递」、「事件传递」来修改或扩展操作系统、应用程序或其他软件组件的行为的各种技术。处理被拦截的函数调用、事件、消息的代码，被称为 钩子（hook） 。 hook 在 PyTorch 中是一个非常有用的特性，利用它可以在不必改变网络输入输出结构的情况下，方便地获取、改变网络中间层变…