快捷導(dǎo)航

使用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)文章

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

pytorch模型訓(xùn)練的時(shí)候GPU使用率不高的問(wèn)題

這篇文章主要介紹了pytorch模型訓(xùn)練的時(shí)候GPU使用率不高的問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

2023-09-09

Django分組聚合查詢實(shí)例分享

在本篇文章里小編給大家分享的是關(guān)于Django分組聚合查詢實(shí)例內(nèi)容，需要的朋友們可以參考下。

2020-04-04

Python爬蟲獲取基金凈值信息詳情

這篇文章主要介紹了Python爬蟲獲取基金凈值信息詳情，文章基于錢兩篇文章的內(nèi)容圍繞python的相關(guān)資料展開(kāi)詳細(xì)介紹，需要的小伙伴可以參考一下

2022-05-05

如何用六步教會(huì)你使用python爬蟲爬取數(shù)據(jù)

網(wǎng)絡(luò)爬蟲就是按照一定規(guī)則自動(dòng)訪問(wèn)互聯(lián)網(wǎng)上的信息并把內(nèi)容下載下來(lái)的程序或腳本,下面這篇文章主要給大家介紹了關(guān)于如何用六步教會(huì)你使用python爬蟲爬取數(shù)據(jù)的相關(guān)資料,需要的朋友可以參考下

2022-04-04