您可以
np.einsum在计算出的差异后使用
broadcastedway,如下所示:
ab = a[:,None,:] - bout = np.einsum('ijk,ijk->ij',ab,ab)
或使用
scipy'scdist其可选的metric参数集,
'sqeuclidean'以根据问题的需要给我们平方的欧几里得距离,如下所示-
from scipy.spatial.distance import cdistout = cdist(a,b,'sqeuclidean')
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)