PyTorch 学习笔记(六):PyTorch hook 和关于 PyTorch backward 过程的理解

在看pytorch官方文档的时候,发现在nn.Module部分和Variable部分均有hook的身影。感到很神奇,因为在使用tensorflow的时候没有碰到过这个词。所以打算一探究竟。

返回顶部