使用pytorch 篩選出一定范圍的值

更新時(shí)間：2020年06月28日 09:59:21 作者：guotong1988

這篇文章主要介紹了使用pytorch 篩選出一定范圍的值，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧

我就廢話不多說了，大家還是直接看代碼吧~

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ǔ)充知識：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)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Python批量處理圖片大小尺寸方法詳解
這篇文章主要介紹了用Python批量處理圖片大小尺寸的方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)吧
2022-11-11
Python?Fuzzywuzzy庫基本函數(shù)及模糊字符串匹配應(yīng)用實(shí)戰(zhàn)
fuzzywuzzy?是一個(gè)用于模糊字符串匹配的?Python?庫,它基于編輯距離算法,提供了多個(gè)函數(shù)來比較字符串之間的相似性,在實(shí)際開發(fā)中,字符串匹配是一項(xiàng)常見但具有挑戰(zhàn)性的任務(wù),用戶可能犯拼寫錯(cuò)誤,使用縮寫或者輸入同義詞,因此,我們需要一種方法來處理這些情況
2023-12-12
這篇文章主要為大家介紹了python?Exception常規(guī)異?；愒斀猓行枰呐笥芽梢越梃b參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪
2023-06-06