У меня есть пример игрушки
a = torch.ones(10)
b = torch.nn.Parameter(a,requires_grad=True)
c = (b**2).sum()
c.backward()
print(b.grad)
print(a.grad)
b.grad, рассчитанный успешно, но a.grad равен None. Как сделать градиент потока через torch.nn.Parameter? Этот пример выглядит искусственно, но я работаю с классом A, полученным из nn.Module, и его параметры инициализируются с выводами из какого-то другого модуля B, и я хочу, чтобы градиенты проходили через параметры A к параметрам B.