标签: python numpy
我有n个d矩阵U和m个d矩阵V,我想计算成对每行与U之间的距离以及V中的每一行。基本上,我想要一个矩阵M,使M[i][j] == np.linalg.norm(U[i] - V[j]) ** 2。我必须做很多计算,所以我想对它进行矢量化。现在,它只是使用2d for循环太慢了。显然,matlab has a way of doing this using pdist2,但我不确定是否有一个numpy等价物。
n
d
U
m
V
M
M[i][j] == np.linalg.norm(U[i] - V[j]) ** 2
pdist2