pytorch计时函数

PyTorch提供了一个计时函数，可以用来测量代码的执行时间。具体使用方法如下：
torch.cuda.synchronize() # 同步GPU和CPU
start_time = time.time() # 记录开始时间
# 执行需要计时的代码
end_time = time.time() # 记录结束时间
print("Time used: {}s".format(end_time - start_time)) # 输出执行时间

其中，torch.cuda.synchronize()用于同步GPU和CPU，确保计时的准确性。start_time和end_time分别记录开始和结束时间，两者相减即为代码执行时间。需要注意的是，计时函数只能用于GPU计算的代码，不能用于CPU计算的代码。