在注册PostBackwardHook时,调用Autograd.Function输入中,既有需要求导的非叶子节点,也有不需要求导的非叶子节点,生成的输出中所有非叶子节点的requires_grad全变成True,不符合预期
该问题是怎么引起的?
在注册PostBackwardHook时,调用Autograd.Function输入中,既有需要求导的非叶子节点,也有不需要求导的非叶子节点,生成的输出中所有非叶子节点的requires_grad全变成True,不符合预期
重现步骤
报错信息