基于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实现对于任意shape的torch.tensor,如果其中的element不等于5则为0，等于5则保留原数值
实现该功能的两种方式，并比较两种实现方式的速度
'''
# 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和布尔索引的速度比较就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
您可能感兴趣的文章:
相关文章
这篇文章主要介绍了python验证多组数据之间有无显著差异，利用方差分析和卡方分布验证多组数据之间的某些属性有无显著性差异，对于连续性属性可以用方差分析，对于离散型属性可以用卡方检验。下面文章详细内容需要的小伙伴可以参考一下2022-01-01
这篇文章主要介绍了Python urllib2运行过程原理解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-06-06
在python中使用requests 模拟浏览器发送请求数据的方法
今天小编就为大家分享一篇在python中使用requests 模拟浏览器发送请求数据的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
最近写了Python函数的功能，犯了一些错误。所以本文主要和大家一起梳理下Python函数的传参机制，感兴趣的小伙伴可以跟随小编一起学习一下2022-07-07
Python中线程threading.Thread的使用详解
python的thread模块是比较底层的模块，python的threading模块是对thread做了一些包装的，可以更加方便的被使用。本文将为大家详细介绍一下python中的线程threading.Thread()的使用，需要的可以参考一下2022-07-07
在本篇文章里小编给大家整理的是一篇关于python文件路径操作方法总结内容，有需要的朋友们可以学习下。2020-12-12
今天小编就为大家分享一篇Pandas实现DataFrame按行求百分数(比例数)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要介绍了python中5个带key的内置函数,包括max取最大值函数，min取最小值函数，filter过滤函数，本文通过示例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-07-07
在性能和资源占用方面,venv 通常更适合轻量级项目、纯Python应用以及对资源敏感的场景,其简单快速的特点使得环境创建和管理更为高效,这篇文章主要介绍了Python虚拟环境conda的安装使用教程,需要的朋友可以参考下2024-03-03
python设置 matplotlib 正确显示中文的四种方式
这篇文章主要介绍了python设置 matplotlib 正确显示中文的四种方式，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下2021-05-05
最新评论