基于torch.where和布爾索引的速度比較

更新時間：2020年01月02日 09:37:06 作者：WYXHAHAHA123

今天小編就為大家分享一篇基于torch.where和布爾索引的速度比較，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

我就廢話不多說了，直接上代碼吧！

import torch
import time
x = torch.Tensor([[1, 2, 3], [5, 5, 5], [7, 8, 9],[5,5,5],[1,2,3,],[1,2,4]])
'''
使用pytorch實現(xiàn)對于任意shape的torch.tensor,如果其中的element不等于5則為0，等于5則保留原數(shù)值
實現(xiàn)該功能的兩種方式，并比較兩種實現(xiàn)方式的速度
'''

# x[x!=5]=1
def t2(x):
  x[x!=5]=0
  return x
def t(x):
  zeros=torch.zeros(x.shape)
  # ones=torch.ones(x.shape)
  x=torch.where(x!=5,zeros,x)
  return x

t2_start=time.time()
t2=t2(x)
t2_end=time.time()

t_start=time.time()
t=t(x)
t_end=time.time()
print(t2,t)
print(torch.sum(t-t2))

print('using x[x!=5]=0 time:',t2_end-t2_start)
print('using torch.where time:',t_end-t_start)
'''
tensor([[0., 0., 0.],
    [5., 5., 5.],
    [0., 0., 0.],
    [5., 5., 5.],
    [0., 0., 0.],
    [0., 0., 0.]]) tensor([[0., 0., 0.],
    [5., 5., 5.],
    [0., 0., 0.],
    [5., 5., 5.],
    [0., 0., 0.],
    [0., 0., 0.]])
tensor(0.)
using x[x!=5]=0 time: 0.0010008811950683594
using torch.where time: 0.0

看來大神說的沒錯，果然是使用torch.where速度更快
 a[a!=5]=0 這種寫法，速度比 torch.where 慢了超級多
'''

以上這篇基于torch.where和布爾索引的速度比較就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

Pytorch 多維數(shù)組運算過程的索引處理方式

相關(guān)文章

python驗證多組數(shù)據(jù)之間有無顯著差異
這篇文章主要介紹了python驗證多組數(shù)據(jù)之間有無顯著差異，利用方差分析和卡方分布驗證多組數(shù)據(jù)之間的某些屬性有無顯著性差異，對于連續(xù)性屬性可以用方差分析，對于離散型屬性可以用卡方檢驗。下面文章詳細內(nèi)容需要的小伙伴可以參考一下
2022-01-01
Python urllib2運行過程原理解析
這篇文章主要介紹了Python urllib2運行過程原理解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2020-06-06
在python中使用requests 模擬瀏覽器發(fā)送請求數(shù)據(jù)的方法
今天小編就為大家分享一篇在python中使用requests 模擬瀏覽器發(fā)送請求數(shù)據(jù)的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-12-12
一文搞懂Python的函數(shù)傳參機制
最近寫了Python函數(shù)的功能，犯了一些錯誤。所以本文主要和大家一起梳理下Python函數(shù)的傳參機制，感興趣的小伙伴可以跟隨小編一起學習一下
2022-07-07
Python中線程threading.Thread的使用詳解
python的thread模塊是比較底層的模塊，python的threading模塊是對thread做了一些包裝的，可以更加方便的被使用。本文將為大家詳細介紹一下python中的線程threading.Thread()的使用，需要的可以參考一下
2022-07-07
python文件路徑操作方法總結(jié)
在本篇文章里小編給大家整理的是一篇關(guān)于python文件路徑操作方法總結(jié)內(nèi)容，有需要的朋友們可以學習下。
2020-12-12
Pandas實現(xiàn)DataFrame按行求百分數(shù)(比例數(shù))
今天小編就為大家分享一篇Pandas實現(xiàn)DataFrame按行求百分數(shù)(比例數(shù))，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-12-12
淺析python中5個帶key的內(nèi)置函數(shù)
這篇文章主要介紹了python中5個帶key的內(nèi)置函數(shù),包括max取最大值函數(shù)，min取最小值函數(shù)，filter過濾函數(shù)，本文通過示例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2022-07-07
Python虛擬環(huán)境conda的安裝使用教程
在性能和資源占用方面,venv?通常更適合輕量級項目、純Python應用以及對資源敏感的場景,其簡單快速的特點使得環(huán)境創(chuàng)建和管理更為高效,這篇文章主要介紹了Python虛擬環(huán)境conda的安裝使用教程,需要的朋友可以參考下
2024-03-03
python設(shè)置 matplotlib 正確顯示中文的四種方式
這篇文章主要介紹了python設(shè)置 matplotlib 正確顯示中文的四種方式，幫助大家更好的理解和學習使用python，感興趣的朋友可以了解下
2021-05-05