Pytorch高階OP操作where,gather原理
PyTorch是一個非常有可能改變深度學(xué)習(xí)領(lǐng)域前景的Python庫。我嘗試使用了幾星期PyTorch,然后被它的易用性所震驚,在我使用過的各種深度學(xué)習(xí)庫中,PyTorch是最靈活、最容易掌握的。
一、where
1)torch.where(condition, x, y) # condition是條件,滿足條件就返回x,不滿足就返回y

2)特點,相比for循環(huán)的優(yōu)點是:可以布置在GPU上運行

二、gather
1)官方解釋:根據(jù)指定的維度和索引值來篩選值

2)舉例

以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
Python restful框架接口開發(fā)實現(xiàn)
這篇文章主要介紹了Python restful框架接口開發(fā)實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-04-04
Python使用Dijkstra算法實現(xiàn)求解圖中最短路徑距離問題詳解
這篇文章主要介紹了Python使用Dijkstra算法實現(xiàn)求解圖中最短路徑距離問題,簡單描述了Dijkstra算法的原理并結(jié)合具體實例形式分析了Python使用Dijkstra算法實現(xiàn)求解圖中最短路徑距離的相關(guān)步驟與操作技巧,需要的朋友可以參考下2018-05-05
Python實現(xiàn)提取谷歌音樂搜索結(jié)果的方法
這篇文章主要介紹了Python實現(xiàn)提取谷歌音樂搜索結(jié)果的方法,涉及Python針對谷歌音樂相關(guān)信息的獲取技巧,具有一定參考借鑒價值,需要的朋友可以參考下2015-07-07
python腳本調(diào)用iftop 統(tǒng)計業(yè)務(wù)應(yīng)用流量的思路詳解
這篇文章主要介紹了python腳本調(diào)用iftop 統(tǒng)計業(yè)務(wù)應(yīng)用流量的思路詳解,本文給大家介紹的非常詳細,具有一定的參考借鑒價值,需要的朋友可以參考下2019-10-10

