2018.11.14——pytorch中的grad_fn,requires_grad

x.grad_fn

x.requires_grad

x.data

x.grad

grad_fn的值可以得知该变量是否是一个计算结果,也就是说该变量是不是一个函数的输出值。每个变量都有grad_fn

代码结果解释:x不是函数输出值,输出:None。a是函数输出值,而且是加法运算。也可以说grad_fn存储运算符号的。具体流程:

----------------------------------------------------------------------------------------------------------------------------------------------------------------------

m = Variable(torch.randn(2,2),requires_grad=True)

表明,需要对哪个变量求导,就设置成requires_grad=True

猜你喜欢

转载自blog.csdn.net/jingtingxu369/article/details/84073191