Update README.md
Browse files
README.md
CHANGED
|
@@ -85,6 +85,7 @@ with torch.no_grad():
|
|
| 85 |
# Calculate similarities
|
| 86 |
scores = torch.matmul(q_reps, p_reps.T)
|
| 87 |
print(scores)
|
|
|
|
| 88 |
```
|
| 89 |
|
| 90 |
# Limitations
|
|
|
|
| 85 |
# Calculate similarities
|
| 86 |
scores = torch.matmul(q_reps, p_reps.T)
|
| 87 |
print(scores)
|
| 88 |
+
# tensor([[-0.0112, 0.3316, 0.2376]], device='cuda:0')
|
| 89 |
```
|
| 90 |
|
| 91 |
# Limitations
|