def loopy_less_app(embed_vec, R):
    tot_vec = embed_vec.shape[0]
Rsq = R ** 2
out = np.zeros(tot_vec, dtype=int)
for i in range(tot_vec):
    subs = embed_vec[i] - embed_vec[i + 1:tot_vec]
dists = np.einsum('ij,ij->i', subs, subs)
out[i] = np.count_nonzero(dists < Rsq)
return out
как эффективно сравнивать записи в массиве numpy друг с другом?