关于view运行速度探究

当只是生成view视图的时候,不用重新分配内存根本不耗费时间。仅仅使用permute的时间其实也就10-4量级,忽略不计。但是加上contiguous()重新分配内存后,耗时将会增加1000倍,估计还会随着数组的大小变化而变化

import torch
import time
s=torch.rand(9,32,160,160)
b,c,h,w=s.size()
st=time.time()
for i in range(40):
    g=s.view(b*c,h,w).permute(0,2,1).contiguous()
ed=time.time()
print(ed-st)
st=time.time()
for i in range(40):
    g=s.view(b*c,h,w)
ed=time.time()
print(ed-st)

 

输出:

0.34734344482421875
0.0002682209014892578
 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值