向量化
对于向量之间的点乘,最开始会想到用 for 循环遍历,但是 for 循环速度很慢。可以在 python 中可以使用向量化(批量化处理)代码来实现快速的计算:
其原理为:第三方库 numpy 能充分利用并行化来加快计算,并行化指令(SIMD指令)在 CPU 和 GPU 上都是允许的。
import numpy as np
c = np.dot(a,b) # 向量点乘
-
关于 numpy 中 dot 函数的使用,可参考CSDN上的博客:
https://blog.csdn.net/g310773517/article/details/139266771
关于 numpy 中其他内置函数的使用,可参考CSDN上的博客:
https://blog.csdn.net/nihaoxiaocui/article/details/51992860
示例:
import numpy as np
import time
a = np.random.rand(1000000)