在?pytorch?中實現(xiàn)計算圖和自動求導(dǎo)
前言:
今天聊一聊 pytorch 的計算圖和自動求導(dǎo),我們先從一個簡單例子來看,下面是一個簡單函數(shù)建立了 yy 和 xx 之間的關(guān)系
然后我們結(jié)點和邊形式表示上面公式:
上面的式子可以用圖的形式表達,接下來我們用 torch 來計算 x 導(dǎo)數(shù),首先我們創(chuàng)建一個 tensor 并且將其requires_grad
設(shè)置為True
表示隨后反向傳播會對其進行求導(dǎo)。
x = torch.tensor(3.,requires_grad=True)
然后寫出
y = 3*x**2 + 4*x + 2
y.backward() x.grad
通過調(diào)用y.backward()
來進行求導(dǎo),這時就可以通過x.grad
來獲得x
的導(dǎo)數(shù)
x.requires_grad_(False)
可以通過requires_grad_
讓x
不參與到自動求導(dǎo)
for epoch in range(3): y = 3*x**2 + 4*x + 2 y.backward() print(x.grad) x.grad.zero_()
如果這里沒有調(diào)用x.grad_zero_()
就是把每次求導(dǎo)數(shù)和上一次求導(dǎo)結(jié)果進行累加。
鏈式法則
相對于 z 對 x 求偏導(dǎo)時,我們可以將 y 看成常數(shù),這樣 x 導(dǎo)數(shù)是 1 那么
x = torch.tensor([1.,2.,3.],requires_grad=True)
y = x * 2 + 3 z = y **2
out = z.mean() out.backward()
print(out) #tensor(51.6667, grad_fn=<MeanBackward0>)
print(x.grad) #tensor([ 6.6667, 9.3333, 12.0000])
對于一個簡單的網(wǎng)絡(luò),我們可以手動計算梯度,但是如果擺在你面前的是一個有152 層的網(wǎng)絡(luò)怎么辦?或者該網(wǎng)絡(luò)有多個分支。這時你的計算復(fù)雜程度可想而知。接下來會帶來更深入自動求導(dǎo)內(nèi)部機制
到此這篇關(guān)于在 pytorch 中實現(xiàn)計算圖和自動求導(dǎo)的文章就介紹到這了,更多相關(guān) pytorch 計算圖 內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
- 使用pytorch進行張量計算、自動求導(dǎo)和神經(jīng)網(wǎng)絡(luò)構(gòu)建功能
- pytorch如何定義新的自動求導(dǎo)函數(shù)
- Pytorch自動求導(dǎo)函數(shù)詳解流程以及與TensorFlow搭建網(wǎng)絡(luò)的對比
- 淺談Pytorch中的自動求導(dǎo)函數(shù)backward()所需參數(shù)的含義
- pytorch中的自定義反向傳播,求導(dǎo)實例
- 關(guān)于PyTorch 自動求導(dǎo)機制詳解
- Pytorch反向求導(dǎo)更新網(wǎng)絡(luò)參數(shù)的方法
- 關(guān)于pytorch求導(dǎo)總結(jié)(torch.autograd)
相關(guān)文章
python回調(diào)函數(shù)中使用多線程的方法
這篇文章主要介紹了python回調(diào)函數(shù)中使用多線程的方法,需要的朋友可以參考下2017-12-12python如何處理matlab的mat數(shù)據(jù)
這篇文章主要介紹了python如何處理matlab的mat數(shù)據(jù),具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2022-05-05Python中g(shù)lob庫實現(xiàn)文件名的匹配
本文主要主要介紹了Python中g(shù)lob庫實現(xiàn)文件名的匹配,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-06-06Python實現(xiàn)解析參數(shù)的三種方法詳解
這篇文章主要介紹了python解析參數(shù)的三種方法,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2022-07-07python如何利用paramiko執(zhí)行服務(wù)器命令
這篇文章主要介紹了python如何利用paramiko執(zhí)行服務(wù)器命令,幫助大家更好的理解和使用python,感興趣的朋友可以了解下2020-11-11Python flask框架實現(xiàn)瀏覽器點擊自定義跳轉(zhuǎn)頁面
這篇文章主要介紹了Python flask框架實現(xiàn)瀏覽器點擊自定義跳轉(zhuǎn)頁面,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-06-06Python requests timeout的設(shè)置
這篇文章主要介紹了Python requests timeout的設(shè)置,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-04-04python編程培訓(xùn) python培訓(xùn)靠譜嗎
現(xiàn)在大家都知道,比較火的編程語言就是python了,很多朋友都想學(xué)習(xí)python編程,想上一個好的python培訓(xùn)班,小編今天給大家全面分析一下關(guān)于python編程培訓(xùn)方面的問題,希望能給你答疑解惑。2018-01-01Python實例方法、類方法、靜態(tài)方法區(qū)別詳解
這篇文章主要介紹了Python實例方法、類方法、靜態(tài)方法區(qū)別詳解,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-09-09