Update README.md (#4)

- Update README.md (ce1f667fb345e885d15ef2289707e1184fdc4aac)


Co-authored-by: Stéphan Tulkens <[email protected]>

README.md

@@ -192,7 +192,7 @@ embeddings = F.normalize(embeddings, p=2, dim=1)
 print(embeddings.shape)
 # torch.Size([2, 768])
 
-similarity = F.cosine_similarity(embeddings[0], embeddings[1])
+similarity = F.cosine_similarity(embeddings[0], embeddings[1], dim=0)
 print(similarity)
 # tensor(0.9118)
 ```