使用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)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了用Python批量處理圖片大小尺寸的方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習吧2022-11-11
Python?Fuzzywuzzy庫基本函數(shù)及模糊字符串匹配應用實戰(zhàn)
fuzzywuzzy?是一個用于模糊字符串匹配的?Python?庫,它基于編輯距離算法,提供了多個函數(shù)來比較字符串之間的相似性,在實際開發(fā)中,字符串匹配是一項常見但具有挑戰(zhàn)性的任務,用戶可能犯拼寫錯誤,使用縮寫或者輸入同義詞,因此,我們需要一種方法來處理這些情況2023-12-12
這篇文章主要介紹了pytorch模型訓練的時候GPU使用率不高的問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
2023-09-09 
在本篇文章里小編給大家分享的是關于Django分組聚合查詢實例內(nèi)容，需要的朋友們可以參考下。
2020-04-04 
這篇文章主要介紹了Python爬蟲獲取基金凈值信息詳情，文章基于錢兩篇文章的內(nèi)容圍繞python的相關資料展開詳細介紹，需要的小伙伴可以參考一下
2022-05-05 
如何用六步教會你使用python爬蟲爬取數(shù)據(jù)
網(wǎng)絡爬蟲就是按照一定規(guī)則自動訪問互聯(lián)網(wǎng)上的信息并把內(nèi)容下載下來的程序或腳本,下面這篇文章主要給大家介紹了關于如何用六步教會你使用python爬蟲爬取數(shù)據(jù)的相關資料,需要的朋友可以參考下
2022-04-04 
這篇文章主要介紹了Django中URL視圖函數(shù)的一些高級概念,Django是Python重多人氣框架中最為著名的一個,需要的朋友可以參考下
2015-07-07 
在本篇文章里小編給大家整理的是一篇關于python爬蟲判斷招聘信息是否存在的實例代碼，有興趣的朋友們可以學習下。
2020-11-11 
Django 解決新建表刪除后無法重新創(chuàng)建等問題
這篇文章主要介紹了Django 解決新建表刪除后無法重新創(chuàng)建等問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-05-05