torch中矢量的计算方式

方法一,使用遍历的方式相加(标量法)

import torch
from time import time

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

# 方法一
start = time()
c = torch.zeros(100)
for i in range(100):
    c[i] = a[i] + b[i]
print(time() - start)

输出为:

'''
0.016082286834716797
'''

方法二：将这两个向量直接做矢量加法。

import torch
from time import time
a = torch.ones(100)
b = torch.ones(100)

start = time()
d = torch.zeros(100)
d = a + b
print(time() - start)

结果:

0.0

所以在使用torch进行向量计算的时候,考虑到时间的因素,尽量使用向量的计算方式比较合适.