import scipy.spatial
import numpy as np
query_embedding = np.array([0,0,1])
corpus_embeddings = np.array([[0,0,1],[1,0,0],[0,1,0]])
distances = scipy.spatial.distance.cdist([query_embedding], corpus_embeddings, "cosine")[0]
print(distances)
print结果:
[0. 1. 1.]