PyTorch-如何获得相对于输入的第一层的渐变?

据我了解,通过使用后向挂钩,索引0处的渐变输入为我提供了相对于输入的渐变。这适用于除第一层以外的所有层。在此索引中,索引0的值为None。

有人可以解释如何在PyTorch中获得对输入图像的梯度逼真度吗?

用于后向挂钩的PyTorch文档和一个示例:https://pytorch.org/docs/stable/nn.html#torch.nn.Module.register_backward_hook https://pytorch.org/tutorials/beginner/former_torchies/nnft_tutorial.html#forward-and-backward-function-hooks

zhangdh_nwnu 回答:PyTorch-如何获得相对于输入的第一层的渐变?

我只是忘记为输入数据设置.requires_grad_()。现在我得到渐变,而不是无。 :D

本文链接:https://www.f2er.com/3143723.html

大家都在问