Мне нужно вычислить градиент выхода сети (не потери) относительно входа, но я получаю ошибку grad can be implicitly created only for scalar outputs.Я пытаюсь воспроизвести то, что описано в этой статье , и мне нужен упомянутый градиент для выполнения шага 7 
Это то, что я хотел быделать:
out = netD(input).view(-1)
out.backward(retain_graph=True)
grad = input.grad