在PyTorch中Tensor的查找和篩選例子
更新時(shí)間：2019年08月18日 14:46:07 作者：玄云飄風(fēng)
今天小編就為大家分享一篇在PyTorch中Tensor的查找和篩選例子，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
本文源碼基于版本1.0，交互界面基于0.4.1
import torch
按照指定軸上的坐標(biāo)進(jìn)行過(guò)濾
index_select()
沿著某tensor的一個(gè)軸dim篩選若干個(gè)坐標(biāo)
>>> x = torch.randn(3, 4) # 目標(biāo)矩陣
>>> x
tensor([[ 0.1427, 0.0231, -0.5414, -1.0009],
[-0.4664, 0.2647, -0.1228, -1.1068],
[-1.1734, -0.6571, 0.7230, -0.6004]])
>>> indices = torch.tensor([0, 2]) # 在軸上篩選坐標(biāo)
>>> torch.index_select(x, dim=0, indices) # 指定篩選對(duì)象、軸、篩選坐標(biāo)
tensor([[ 0.1427, 0.0231, -0.5414, -1.0009],
[-1.1734, -0.6571, 0.7230, -0.6004]])
>>> torch.index_select(x, dim=1, indices)
tensor([[ 0.1427, -0.5414],
[-0.4664, -0.1228],
[-1.1734, 0.7230]])
where()
用于將兩個(gè)broadcastable的tensor組合成新的tensor，類似于c++中的三元操作符“?:”
>>> x = torch.randn(3, 2)
>>> y = torch.ones(3, 2)
>>> torch.where(x > 0, x, y)
tensor([[1.4013, 1.0000],
[1.0000, 0.9267],
[1.0000, 0.4302]])
>>> x
tensor([[ 1.4013, -0.9960],
[-0.3715, 0.9267],
[-0.7163, 0.4302]])
指定條件返回01-tensor
>>> x = torch.arange(5) >>> x tensor([0, 1, 2, 3, 4]) >>> torch.gt(x,1) # 大于 tensor([0, 0, 1, 1, 1], dtype=torch.uint8) >>> x>1 # 大于 tensor([0, 0, 1, 1, 1], dtype=torch.uint8) >>> torch.ne(x,1) # 不等于 tensor([1, 0, 1, 1, 1], dtype=torch.uint8) >>> x!=1 # 不等于 tensor([1, 0, 1, 1, 1], dtype=torch.uint8) >>> torch.lt(x,3) # 小于 tensor([1, 1, 1, 0, 0], dtype=torch.uint8) >>> x<3 # 小于 tensor([1, 1, 1, 0, 0], dtype=torch.uint8) >>> torch.eq(x,3) # 等于 tensor([0, 0, 0, 1, 0], dtype=torch.uint8) >>> x==3 # 等于 tensor([0, 0, 0, 1, 0], dtype=torch.uint8)
返回索引
>>> x = torch.arange(5)
>>> x # 1維
tensor([0, 1, 2, 3, 4])
>>> torch.nonzero(x)
tensor([[1],
[2],
[3],
[4]])
>>> x = torch.Tensor([[0.6, 0.0, 0.0, 0.0],[0.0, 0.4, 0.0, 0.0],[0.0, 0.0, 1.2, 0.0],[0.0, 0.0, 0.0,-0.4]])
>>> x # 2維
tensor([[ 0.6000, 0.0000, 0.0000, 0.0000],
[ 0.0000, 0.4000, 0.0000, 0.0000],
[ 0.0000, 0.0000, 1.2000, 0.0000],
[ 0.0000, 0.0000, 0.0000, -0.4000]])
>>> torch.nonzero(x)
tensor([[0, 0],
[1, 1],
[2, 2],
[3, 3]])
借助nonzero()我們可以返回符合某一條件的index（https://stackoverflow.com/questions/47863001/how-pytorch-tensor-get-the-index-of-specific-value）
>>> x=torch.arange(12).view(3,4)
>>> x
tensor([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]])
>>> (x>4).nonzero()
tensor([[1, 1],
[1, 2],
[1, 3],
[2, 0],
[2, 1],
[2, 2],
[2, 3]])
以上這篇在PyTorch中Tensor的查找和篩選例子就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
在tensorflow中設(shè)置使用某一塊GPU、多GPU、CPU的操作
今天小編就為大家分享一篇在tensorflow中設(shè)置使用某一塊GPU、多GPU、CPU的操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-02-02
Python bytes string相互轉(zhuǎn)換過(guò)程解析
這篇文章主要介紹了Python bytes string相互轉(zhuǎn)換過(guò)程解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-03-03
Python中rasterio庫(kù)的實(shí)現(xiàn)
Rasterio是一個(gè)用于讀寫光柵數(shù)據(jù)的Python庫(kù),它專注于快速和可靠地處理大規(guī)模地理空間柵格數(shù)據(jù),本文主要介紹了Python中rasterio庫(kù)的實(shí)現(xiàn),感興趣的可以了解一下2025-04-04
Python之關(guān)于類變量的兩種賦值區(qū)別詳解
這篇文章主要介紹了Python之關(guān)于類變量的兩種賦值區(qū)別詳解,具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-03-03
python中的classmethod與staticmethod
這篇文章主要介紹了python中的classmethod與staticmethod，2022-01-01
Python標(biāo)準(zhǔn)庫(kù)json模塊和pickle模塊使用詳解
這篇文章主要介紹了Python標(biāo)準(zhǔn)庫(kù)json模塊和pickle模塊使用詳解,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-03-03
pycharm使用Translation插件實(shí)現(xiàn)翻譯功能
PyCharm是一款很流行的Python編輯器,經(jīng)常遇到在PyCharm中把中文翻譯成英文的需求,下面這篇文章主要給大家介紹了關(guān)于pycharm使用Translation插件實(shí)現(xiàn)翻譯功能的相關(guān)資料,需要的朋友可以參考下2023-05-05
這篇文章主要介紹了 Python OpenCV讀取顯示視頻的方法示例,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-02-02
Django利用Channels+websocket開發(fā)聊天室完整案例
Channels是Django團(tuán)隊(duì)研發(fā)的一個(gè)給Django提供websocket支持的框架,使用它我們可以輕松開發(fā)需要長(zhǎng)鏈接的實(shí)時(shí)通訊應(yīng)用,下面這篇文章主要給大家介紹了關(guān)于Django利用Channels+websocket開發(fā)聊天室的相關(guān)資料,需要的朋友可以參考下2023-06-06
Python實(shí)現(xiàn)任意文件查找工具的不同方法詳解
在日常的開發(fā)和運(yùn)維工作中,經(jīng)常需要查找特定類型的文件或帶有特定模式的文件,本文將提供兩種不同的編寫文件查找工具實(shí)現(xiàn)方式,需要的可以參考下2025-03-03