register_forward_hook

Torch Hook（钩子）

register_forward_hook（）

主要的作用是在不改变torch网络的情况下获取某一层的输出

步骤：

1. 对于网络中的某一个层使用register_forward_hook方法，方法中传入一个func函数，包裹了model，input，outpu。
2. 函数回返回一个handle，存在remove方法可以删除hook。