技術標籤：Pytorch深度學習pytorch函式求導

剛開始學習pytorch我們先用該框架中的自動求導函式對一個一元二次方程求導來初步體驗一下pytorch在求解偏導的方便性。下面直接上程式碼：

import  torch
from    torch import autograd


x = torch.tensor(1.)
a = torch.tensor(1., requires_grad=True)#對方程係數進行賦初值
b = torch.tensor(2., requires_grad=True)
c = torch.tensor(3., requires_grad=True
 
)

y = a**2 * x + b * x + c

print('before:', a.grad, b.grad, c.grad)
grads = autograd.grad(y, [a, b, c])#利用函式自動求偏導
print('after :', grads[0], grads[1], grads[2])

輸出結果為：