PyTorch自動求導

import torch
from torch import autograd

x=torch.tensor(1.)
a=torch.tensor(1.,requires_grad=True)  #注意帶入的值是float型，有個點。
b=torch.tensor(2.,requires_grad=True)  #requires_grad需要求導
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])   #y分別對a,b,c求導
print("after",grads[0],grads[1],grads[2])  #計算後，直接帶進去需要的值

輸出

注意放的是 float型