Update README.md
Browse files
README.md
CHANGED
@@ -50,7 +50,6 @@ embeddings_tensor = torch.tensor(embeddings)
|
|
50 |
similarities = torch.nn.functional.cosine_similarity(embeddings_tensor.unsqueeze(0), embeddings_tensor.unsqueeze(1), dim=2)
|
51 |
|
52 |
print(similarities)
|
53 |
-
# tensor([[1.0000, 0.
|
54 |
-
# [0.
|
55 |
-
# [0.
|
56 |
-
```
|
|
|
50 |
similarities = torch.nn.functional.cosine_similarity(embeddings_tensor.unsqueeze(0), embeddings_tensor.unsqueeze(1), dim=2)
|
51 |
|
52 |
print(similarities)
|
53 |
+
# tensor([[1.0000, 0.8608, 0.6609],
|
54 |
+
# [0.8608, 1.0000, 0.7046],
|
55 |
+
# [0.6609, 0.7046, 1.0000]])```
|
|