Compute squared L2 distance matrix

l2Distance.py

# dataset [n x d]

X = np.array(data_set)
Y = np.array([[93,  7], [50, 34], [51, 79]])

def l2-Distance(X, MU):
	x = np.sum(X**2, axis=1, keepdims=True)
    y = np.sum(MU**2, axis=1)
    xy = np.dot(X, MU.T)
   
    dist = np.sqrt(x - 2*xy + y) 
    return np.argmin(dist, axis=1)