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上運(yùn)行

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

2)舉例

以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
Python通過解析網(wǎng)頁實現(xiàn)看報程序的方法
這篇文章主要介紹了Python通過解析網(wǎng)頁實現(xiàn)看報程序的方法,比較實用的功能,需要的朋友可以參考下2014-08-08
Tensorflow中k.gradients()和tf.stop_gradient()用法說明
這篇文章主要介紹了Tensorflow中k.gradients()和tf.stop_gradient()用法說明,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
pandas?dataframe獲取所有行名稱與列名稱方法示例
這篇文章主要給大家介紹了關(guān)于pandas?dataframe獲取所有行名稱與列名稱的相關(guān)資料,Pandas是Python中用于數(shù)據(jù)分析的非常重要的庫,它提供了多種方法來獲取列名,需要的朋友可以參考下2023-09-09
在pandas中遍歷DataFrame行的實現(xiàn)方法
這篇文章主要介紹了在pandas中遍歷DataFrame行的實現(xiàn)方法,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-10-10

