首页 > 解决方案 > 将值传递给下一个正向计算

问题描述

我目前正在 pytorch 中构建自己的 autograd 函数(出于科学目的)。

对于我想要实现的目标,我想从我的前锋中将一个值传递给下一个前锋呼叫。所以本质上我正在寻找类似的功能,如 ctx.save_for_backward() 但不是为后向传递保存一个值,而是为下一个前向传递保存一个值,这样我的前向传递可以使用一个值计算前一个前向传递。

pytorch 中是否有这样的功能,或者您将如何处理?

感谢您对这个问题的任何帮助

标签: pythonpytorchautograd

解决方案


推荐阅读