使用pytorch 篩選出一定范圍的值
更新時(shí)間：2020年06月28日 09:59:21 作者：guotong1988
這篇文章主要介紹了使用pytorch 篩選出一定范圍的值，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
我就廢話不多說(shuō)了，大家還是直接看代碼吧~
import torch input_tensor = torch.tensor([1,2,3,4,5]) print(input_tensor>3) mask = (input_tensor>3).nonzero() print(mask) print(input_tensor.index_select(0,mask))
tensor([0, 0, 0, 1, 1], dtype=torch.uint8) tensor([3, 4]) tensor([4, 5])
補(bǔ)充知識(shí)：pytorch tensor篩選滿足條件的行或列（使用與或）
我就廢話不多說(shuō)了，大家還是直接看代碼吧~
import torch x = torch.linspace(1, 8, steps=8).view(4, 2) print(x) area1=(x[:,0]>5.5)&(x[:,1]>5.5) c=x[:,0]*x[:,1] area2=c>25 area=area1|area2 print(x[area]) if 0: # index=torch.max(area,1)[0] b=x[area] # b= x[torch.where((x[:,0]>0) & (x[:,0]<6))] # print(b)
以上這篇使用pytorch 篩選出一定范圍的值就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
對(duì)python同一個(gè)文件夾里面不同.py文件的交叉引用方法詳解
今天小編就為大家分享一篇對(duì)python同一個(gè)文件夾里面不同.py文件的交叉引用方法詳解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-12-12
這篇文章主要介紹了Python匿名函數(shù)，在python中，除了一般使用def定義的函數(shù)外，還有一種使用lambda定義的匿名函數(shù)，這種函數(shù)可以用在任何普通函數(shù)可以使用的地方，但在定義時(shí)被嚴(yán)格限定為單一表達(dá)式，下面來(lái)看詳細(xì)內(nèi)容吧2021-11-11
時(shí)間序列預(yù)測(cè)中的數(shù)據(jù)滑窗操作實(shí)例(python實(shí)現(xiàn))
滑動(dòng)窗口操作非常普遍,非常有用,它們也很容易在Python中實(shí)現(xiàn),下面這篇文章主要給大家介紹了關(guān)于時(shí)間序列預(yù)測(cè)中的數(shù)據(jù)滑窗操作python實(shí)現(xiàn)的相關(guān)資料,需要的朋友可以參考下2022-03-03
這篇文章主要介紹了python文件與路徑管理方法，文章屬于python使用技巧的分享，下文圍繞文件與路徑管理相關(guān)內(nèi)容，需要的小伙伴可以參考一下，希望對(duì)你有所幫助2022-02-02
這篇文章主要介紹了舉例講解Python中的死鎖、可重入鎖和互斥鎖,盡管線程編程方面Python的GIL問(wèn)題老生常談...需要的朋友可以參考下2015-11-11
Python使用ffmpeg合成視頻、音頻的實(shí)現(xiàn)方法
這篇文章主要介紹了Python使用ffmpeg合成視頻、音頻,通過(guò)本文的學(xué)習(xí)能幫助大家了解如何在python中調(diào)用ffmpeg模塊，對(duì)此進(jìn)行音視頻合并，完成視頻合成，需要的朋友可以參考下2022-04-04
超詳細(xì)注釋之OpenCV dlib實(shí)現(xiàn)人臉采集
這篇文章主要介紹了OpenCV dlib實(shí)現(xiàn)人臉采集，本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-09-09
淺談python出錯(cuò)時(shí)traceback的解讀
這篇文章主要介紹了淺談python出錯(cuò)時(shí)traceback的解讀，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-07-07
tensorflow2 自定義損失函數(shù)使用的隱藏坑
本文主要介紹了tensorflow2 自定義損失函數(shù)使用的隱藏坑，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-07-07
這篇文章主要介紹了在Pandas中更改DataFrame中的值方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2023-08-08