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

可以使用 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