hook钩子函数使用

qq_419203233 个月前
pytorch·hook钩子函数使用
pytorch中的hook机制register_forward_hook上篇文章主要介绍了hook钩子函数的大致使用流程，本篇文章主要介绍pytorch中的hook机制register_forward_hook，手动在forward之前注册hook，hook在forward执行以后被自动执行。