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型

来源：CSDN

作者：林灵会灭

链接：https://blog.csdn.net/qq_35159009/article/details/103852000