小编use*_*358的帖子

Python在大矩阵中找到相似向量的最快方法

我正在尝试编写一个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进行这种计算吗?

python numpy raspberry-pi3

5
推荐指数
1
解决办法
2633
查看次数

标签 统计

numpy ×1

python ×1

raspberry-pi3 ×1