PyTorch入门学习 12-矢量计算比循环计算快很多、time()

avatar 2024年04月16日14:59:13 0 188 views
博主分享免费Java教学视频,B站账号:Java刘哥 ,长期提供技术问题解决、项目定制:本站商品点此

两个向量相加或者其他运算,尽可能采用矢量计算。比对每个元素循环计算快几个数量级(几百倍、千倍)。

可以使用 time() 获取当前时间戳毫秒数

代码

from time import time

import torch

a = torch.ones(100000)
b = torch.ones(100000)
c = torch.zeros(100000)

# 例1、按元素循环相加,计算时间
start = time()
for i in range(100000):
    c[i] = a[i] + b[i]

print('耗时:{} ms'.format(time() - start))  # 0.6349880695343018 ms

# 例2、直接相加,计算时间,快很多
start = time()
d = a + b
print('耗时:{} ms'.format(time() - start))  # 0.0002300739288330078ms

 

运行结果

耗时:0.6324949264526367 ms
耗时:0.0003371238708496094 ms

 

  • 微信
  • 交流学习,服务定制
  • weinxin
  • 个人淘宝
  • 店铺名:言曌博客咨询部

  • (部分商品未及时上架淘宝)
avatar

发表评论

avatar 登录者:匿名
匿名评论,评论回复后会有邮件通知

  

已通过评论:0   待审核评论数:0