如何使用python来计算程序运行时间

使用python来统计程序运行时间

  在python中，time是用来统计程序运行时间的包，首先先导入time包。

from time import time

  然后，先在需要计算运行时间的代码片段前面，使用变量start储存下当前时间。

start = time()

  再在运行完的程序片段最后，统计当前时间，和start变量两者作差即可。

from time import time

start = time()
#中间这里是
#需要统计运行时间的代码片段
print(time() - start)

下面使用一个简单的pytorch实例

import torch
from time import time

a = torch.ones(1000)
b = torch.ones(1000)

start = time()
c = torch.zeros(1000)
for i in range(1000):
    c[i] = a[i] + b[i]
print(time() - start) 

运行的结果是：

0.007939338684082031

代码参考自：动手学深度学习 pytorch版