Как рассчитать градиенты на тензор в PyTorch?

Question

Я хочу вычислить градиент тензора и, тем не менее, он дает ошибку как

RunTimeerror: grad can be implicitly created only for scalar outputs

и вот что я пытаюсь кодировать:

x = torch.full((2,3), 4,requires_grad=True)
y = (2*x**2+3)
y.backward()

И в этот момент выдает ошибку.

Answer 1

Поскольку нет суммирования / уменьшения значения потерь, как .sum() Следовательно, проблема может быть решена с помощью:

y.backward(torch.ones_like(x))

, который выполняет произведение вектора Якоби с тензором всех единиц.и получить градиент.