在看pytorch官方文档的时候，发现在nn.Module部分和Variable部分均有hook的身影。感到很神奇，因为在使用tensorflow的时候没有碰到过这个词。所以打算一探究竟。
文章目录 [隐藏]
- 1 Variable 的 hook
- 1.1 register_hook(hook)
- 2 nn.Module的hook
- 2.1 register_forward_hook(hook)
- 3 register_backward_hook
返回顶部
在看pytorch官方文档的时候，发现在nn.Module部分和Variable部分均有hook的身影。感到很神奇，因为在使用tensorflow的时候没有碰到过这个词。所以打算一探究竟。
文章目录 [隐藏]
