Updated w. similarity calculation

README.md

@@ -112,4 +112,10 @@ q_reps_norm = torch.nn.functional.normalize(q_reps, p=2, dim=1)
 d_reps_norm = torch.nn.functional.normalize(d_reps, p=2, dim=1)
 cos_sim = torch.mm(q_reps_norm, d_reps_norm.transpose(0, 1))
 
+print(cos_sim)
+"""
+tensor([[0.7161, 0.4021],
+        [0.4525, 0.7370]])
+"""
+
 ```