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)
# 方法二,省时,直接矢量相加
start = time()
d = a + b
print(time() - start)
李沐深度学习-矢量计算表达式
最新推荐文章于 2024-07-21 07:50:37 发布