基于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学习笔记之读取文件、OS模块、异常处理、with as语法示例
这篇文章主要介绍了Python学习笔记之读取文件、OS模块、异常处理、with as语法,结合实例形式简单分析了Python针对文件、目录的读取、异常处理等相关操作技巧,需要的朋友可以参考下2019-06-06使用Python通过win32 COM打开Excel并添加Sheet的方法
今天小编就为大家分享一篇使用Python通过win32 COM打开Excel并添加Sheet的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
最新评论