python - 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