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_ () для моих входных данных. Теперь я получаю градиенты, а не None. : D