高阶梯度共1篇
排序
让 PyTorch 自定义 Function 支持二次反向传播-未完纪

让 PyTorch 自定义 Function 支持二次反向传播

从保存输入和输出，到把中间结果作为辅助输出，再为无法自动追踪的反向过程定义新的 Function，完整理解 PyTorch 二次反向传播的计算图边界。
技术译文
彩雨伞的头像-未完纪钻石会员彩雨伞3小时前
0340