Python的有趣用法

1、numpy不用循环求两个矩阵中两两向量的欧式距离

#arr1: (N, m).
#arr2: (M, m).
#dis:  (N,M).
dis =np.sqrt( ((arr1[:, None, :] - arr2[None, :, :]) ** 2).sum(-1) )

猜你喜欢

转载自blog.csdn.net/qq_41043389/article/details/127999143