1. 程式人生 > >2018.11.14——pytorch中的grad_fn,requires_grad

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