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