标签: python arrays numpy vectorization numpy-broadcasting
我有两个numpy数组A(500,50)和B(100,50),我想计算A的每一行与B的所有行之间的差异,例如dist(i,j)是第i行之间的差异B的A和第j行。我希望使用广播进行矢量化实现并且没有循环。
下面给出了使用两个循环的实现:
for i in range(A.shape[0]): for j in range(B.shape[0]) dist[i][j]=A[i]-B[j]