PyTorch模块如何进行后部支撑
PyTorch模块如何进行后部支撑
提问于 2018-04-01 12:39:38
在遵循extending PyTorch - adding a module的说明时，我注意到在扩展Module时，我们实际上并不需要实现向后的函数。我们唯一需要做的就是在forward函数中应用Function实例，当做back prop时，PyTorch可以自动调用function实例中的backward函数。这在我看来就像是魔术，因为我们甚至没有注册我们使用的函数实例。我查看了源代码，但没有发现任何相关内容。有没有人能给我指出所有这些事情真正发生的地方？
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/49594858
复制相关文章
相似问题