【深度学习】矢量化加速的Python代码实现（定义计时器）

深藏阁楼爱情的钟 2023-09-27 11:51 91阅读 0赞

由于在深度学习中频繁进行运行时间的基准测试，所以我们定义一个计时器，Python代码实现如下：

### 一、导入需要的包 ###

import math
    import time
    import numpy as np
    import torch
    from d2l import torch as d2l

### 二、定义计时器 ###

class Timer:
        # 记录多次运行时间
        def __init__(self):
            self.times=[]
            self.start()
        def start(self):
            # 启动计时器
            self.tik=time.time()
        def stop(self):
            # 停止计时器并记录时间到列表中
            self.times.append(time.time()-self.tik)
            return self.times[-1]
        def avg(self):
            # 返回平均时间
            return sum(self.times)/len(self.times)
        def sum(self):
            # 返回时间总和
            return sum(self.times)
        def cumsum(self):
            # 返回累计时间
            return np.array(self.times).cumsum().tolist()

### 三、举例测试计时器 ###

n=10000
    a=torch.ones(n)
    b=torch.ones(n)
    c=torch.zeros(n)
    timer=Timer()
    for i in range(n):
        c[i]=a[i]+b[i]
    seconds=f'{
          timer.stop():.5f} seconds'
    print(seconds)

结果展示为：

0.10235 seconds

发表评论取消回复

表情：

评论列表（有 0 条评论，91人围观）

还没有评论，来说两句吧...

相关阅读

相关使用Numba加速Python代码

使用Numba加速Python代码 Python是一种非常流行的编程语言，由于其简单易学和灵活性，被广泛用于科学计算、机器学习、数据分析等领域。然而，Python的执行效率并

青旅半醒/ 2024年03月25日 23:03/ 0 赞/ 52 阅读

相关 Python代码优化实践：使用装饰器实现一个高效的代码计时器

Python代码优化实践：使用装饰器实现一个高效的代码计时器在编写Python程序的过程中，我们经常需要对代码的性能进行优化。其中一个关键指标是运行时间。为了更好地评估程序

落日映苍穹つ/ 2024年03月24日 09:02/ 0 赞/ 28 阅读

相关一些Python代码加速工具！

这篇文章会提供一些优化代码的工具。会让代码变得更简洁，或者更迅速。当然这些并不能代替算法设计，但是还是能让 Python 加速很多倍。其实前面讲算法的文章，也有提到过。

矫情吗；*/ 2023年10月15日 12:41/ 0 赞/ 10 阅读

相关【深度学习】矢量化加速的Python代码实现（定义计时器）

由于在深度学习中频繁进行运行时间的基准测试，所以我们定义一个计时器，Python代码实现如下：一、导入需要的包 import math import

深藏阁楼爱情的钟/ 2023年09月27日 11:51/ 0 赞/ 92 阅读

相关用Cython加速Python代码

安装Cython pip install Cython 如何使用要在我们的笔记本中使用Cython，我们将使用IPython magic命令。Magic

￡神魔★判官ぃ/ 2023年08月17日 16:31/ 0 赞/ 106 阅读

相关实时深度学习的推理加速

还未完成的。。。。。。作者 Yanchen 毕业于普林斯顿大学机器学习方向，现就职于微软Redmond总部，从事大规模分布式机器学习和企业级AI研发工作。在该篇文章中，作

深藏阁楼爱情的钟/ 2022年12月28日 14:09/ 0 赞/ 164 阅读

相关深度学习训练加速--分布式

初稿进行中。。。。一、内部方法 1. 网络结构的选择比如 CNN 与 RNN，前者更适合并行架构 2. 优化算法的改进：动量、自适应学习率 3.

约定不等于承诺〃/ 2022年12月28日 14:02/ 0 赞/ 220 阅读

相关深度学习模型压缩加速

目录前言深度学习模型压缩加速原理和方法为什么需要模型压缩和加速？深度学习模型压缩的方法前端压缩的方法

痛定思痛。/ 2022年12月21日 11:00/ 0 赞/ 175 阅读

相关 GPS数据矢量化JAVA_SVGX矢量化图形编辑器，100%JAVA实现的矢量化图形编辑器

SVGX矢量化图形编辑器，100%JAVA实现的矢量化图形编辑器 SVGX矢量化图形编辑器是面向工程应用的矢量图形制作软件，基于著名的Eclipse GEF图形编辑框架实现了

Bertha 。/ 2022年11月08日 05:55/ 0 赞/ 215 阅读

相关 Alluxio 助力 Kubernetes，加速云端深度学习

![头图.png][.png] 作者 | 车漾阿里云高级技术专家范斌 Alluxio 创始成员，开源社区副总裁来源 | [阿里巴巴云原生公众号][Li

水深无声/ 2022年11月06日 06:54/ 0 赞/ 242 阅读