PyTorch - 如何获得第一层相对于输入的梯度？

Question

根据我的理解，通过使用反向挂钩，索引 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

Answer 1

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