Pytorch高階OP操作where,gather原理
PyTorch是一個(gè)非常有可能改變深度學(xué)習(xí)領(lǐng)域前景的Python庫。我嘗試使用了幾星期PyTorch,然后被它的易用性所震驚,在我使用過的各種深度學(xué)習(xí)庫中,PyTorch是最靈活、最容易掌握的。
一、where
1)torch.where(condition, x, y) # condition是條件,滿足條件就返回x,不滿足就返回y
2)特點(diǎn),相比for循環(huán)的優(yōu)點(diǎn)是:可以布置在GPU上運(yùn)行
二、gather
1)官方解釋:根據(jù)指定的維度和索引值來篩選值
2)舉例
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
python實(shí)現(xiàn)簡單的購物程序代碼實(shí)例
這篇文章主要介紹了python實(shí)現(xiàn)簡單的購物程序代碼實(shí)例,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-03-03Python通過解析網(wǎng)頁實(shí)現(xiàn)看報(bào)程序的方法
這篇文章主要介紹了Python通過解析網(wǎng)頁實(shí)現(xiàn)看報(bào)程序的方法,比較實(shí)用的功能,需要的朋友可以參考下2014-08-08Tensorflow中k.gradients()和tf.stop_gradient()用法說明
這篇文章主要介紹了Tensorflow中k.gradients()和tf.stop_gradient()用法說明,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06pandas?dataframe獲取所有行名稱與列名稱方法示例
這篇文章主要給大家介紹了關(guān)于pandas?dataframe獲取所有行名稱與列名稱的相關(guān)資料,Pandas是Python中用于數(shù)據(jù)分析的非常重要的庫,它提供了多種方法來獲取列名,需要的朋友可以參考下2023-09-09在pandas中遍歷DataFrame行的實(shí)現(xiàn)方法
這篇文章主要介紹了在pandas中遍歷DataFrame行的實(shí)現(xiàn)方法,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-10-10Python算法應(yīng)用實(shí)戰(zhàn)之隊(duì)列詳解
隊(duì)列是一種先進(jìn)先出(First-In-First-Out,F(xiàn)IFO)的數(shù)據(jù)結(jié)構(gòu)。隊(duì)列被用在很多地方,比如提交操作系統(tǒng)執(zhí)行的一系列進(jìn)程、打印任務(wù)池等,一些仿真系統(tǒng)用隊(duì)列來模擬銀行或雜貨店里排隊(duì)的顧客。下面就介紹了Python中隊(duì)列的應(yīng)用實(shí)戰(zhàn),需要的可以參考。2017-02-02