d4852d9 c746c39
1
2
3
4
5
6
7
8
import numpy as np def normalize_embeddings(embeddings): embeddings_norm = np.linalg.norm(embeddings, axis=1) nonzero_embeddings = embeddings_norm > 0.0 embeddings[nonzero_embeddings] /= embeddings_norm[nonzero_embeddings, None] return embeddings
