我正在尝试编写一个Python函数,它接受一个向量(1x128),然后在一个大的未排序矩阵(2000x128)中找到最相似的列。该函数在应用程序中调用了约 100000 次。当我在台式电脑上工作时没有问题,但在树莓派上工作速度非常慢。这是我的功能;
def find_similar_index(a):
d = []
norma=np.linalg.norm(a)
for i in range(0, 1999):
d.append(np.abs(np.linalg.norm(a - A[:, i]))/norma)
return np.argmin(d)
Run Code Online (Sandbox Code Playgroud)
我可以改进此功能中的任何内容以加快工作速度吗?我可以使用Raspberry Pi的GPU进行这种计算吗?