Pytorch獲取無梯度TorchTensor中的值
更新時間：2021年05月22日 08:59:23 作者：_Epsilon_
這篇文章主要介紹了Pytorch獲取無梯度TorchTensor中的值，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
獲取無梯度Tensor
遇到的問題：
使用兩個網絡并行運算，一個網絡的輸出值要給另一個網絡反饋。而反饋的輸出值帶有網絡權重的梯度，即grad_fn=<XXXBackward0>.
這時候如果把反饋值扔到第二網絡中更新，會出現(xiàn)第一個計算圖丟失無法更新的錯誤。哎喲喂，我根本不需要第一個網絡的梯度好嗎？
一開始用了一個笨辦法，先轉numpy，然后再轉回torch.Tensor。因為numpy數(shù)據是不帶梯度的。
但是我的原始tensor的放在cuda上的，
cuda的張量是不能直接轉Tensor，所以
t_error = td_error.cuda().data.cpu().numpy() t_error = torch.FloatTensor(t_error).to(device)
從cuda轉回了cpu，變成numpy，又轉成了tensor，又回到了cuda上，坑爹呢這是，可能只有我才能寫出如此低效的辣雞代碼了。
后來發(fā)現(xiàn)，其實直接在返回的時候添加
with torch.no_grad(): td_error = reward + GAMMA * v_ - v
即可.
補充：在pytorch中取一個tensor的均值，然后該張量中的所有值與其對比！
Pytorch中的Tensor的shape是(B, C, W, H)，
對該tensor取均值并與所有值做對比代碼如下：
C, H, W = tensor.shape[1], tensor.shape[2], tensor.shape[3] for c in range(C): mean = torch.mean(x[0][c]) for h in range(H): for w in range(W): if x[0][c][h][w] >= mean: x[0][c][h][w] = mean
以上為個人經驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
Python實現(xiàn)二分法查找及優(yōu)化的示例詳解
二分查找法（Binary?Search）是一種在有序數(shù)組中查找某一特定元素的算法，在本文中，我們將使用?Python?實現(xiàn)二分查找算法，并深入探討算法的原理和實現(xiàn)細節(jié)，感興趣的可以了解一下2023-04-04
詳解使用Pytorch Geometric實現(xiàn)GraphSAGE模型
這篇文章主要為大家介紹了詳解使用Pytorch Geometric實現(xiàn)GraphSAGE模型示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2023-04-04
這篇文章主要介紹了使用Python處理EXCEL基礎操作篇1,如何在Python中導入EXCEL數(shù)據,文中提供了解決思路和部分實現(xiàn)代碼，一起來看看吧2023-03-03
這篇文章主要介紹了深入解析Python中的WSGI接口,WSGI接口是Python中網絡框架連接服務器的必備工具,需要的朋友可以參考下2015-05-05
在本文中小編給大家整理了關于Python函數(shù)和模塊的使用的相關知識點以及實例代碼內容，需要的朋友們跟著學習下。2019-05-05
今天小編就為大家分享一篇將pytorch轉成longtensor的簡單方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
python3+PyQt5 創(chuàng)建多線程網絡應用-TCP客戶端和TCP服務器實例
今天小編就為大家分享一篇python3+PyQt5 創(chuàng)建多線程網絡應用-TCP客戶端和TCP服務器實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-06-06
在腳本或應用程序中,我們需要執(zhí)行音頻處理任務,下面這篇文章主要給大家介紹了關于Python如何對音視頻文件進行解析的相關資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下2022-11-11
這篇文章主要介紹了Python批量裁剪圖片的程序代碼，是批量裁剪某一文件夾下的所有圖片，并指定裁剪寬高，本文給大家分享實現(xiàn)思路,需要的朋友可以參考下2022-07-07
今天教大家如何學會通過python的matplotlib庫繪圖,文中有非常詳細的圖文解說及代碼示例,對正在學習python的小伙伴們很有幫助,需要的朋友可以參考下2021-05-05