NumPy Yayını: İki dizi arasındaki karesel farkların toplamının hesaplanması

Question

Aşağıdaki koda sahibim. Python'da sonsuza dek sürüyor.

ab = a[:,None,:] - b 
out = np.einsum('ijk,ijk->ij',ab,ab) 

Veya onun opsiyonel scipy's cdist kullanmak -

def euclidean_square(a,b): 
    squares = np.zeros((a.shape[0],b.shape[0])) 
    for i in range(squares.shape[0]): 
     for j in range(squares.shape[1]): 
      diff = a[i,:] - b[j,:] 
      sqr = diff**2.0 
      squares[i,j] = np.sum(sqr) 
    return squares 

Answer 1

Öyle gibi bir broadcasted way farklılıkları hesapladıktan sonra np.einsum kullanabilirsiniz ... Yayın içine bu hesaplamayı çevirmek için bir yolu olmalı bizim sorunu için gerektiği gibi 'sqeuclidean' olarak ayarlanmış metrik argüman şöyle bize kare öklid mesafeleri vermek -

from scipy.spatial.distance import cdist 
out = cdist(a,b,'sqeuclidean') 

Answer 2

cdist kullanarak yanında başka çözüm aşağıdaki

difference_squared = np.zeros((a.shape[0], b.shape[0])) 
for dimension_iterator in range(a.shape[1]): 
    difference_squared = difference_squared + np.subtract.outer(a[:, dimension_iterator], b[:, dimension_iterator])**2.