使用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版