Update README.md

unlimblue · May 21, 2019 · eef4222 · eef4222
1 parent 1a252cf
commit eef4222
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -73,5 +73,5 @@ knn = KNN(k=10, transpose_mode=True)
 ref = torch.rand(32, 1000, 5).cuda()
 query = torch.rand(32, 50, 5).cuda()
 
-dist, indx = knn(ref, query)  # 32 x 50 x 5
+dist, indx = knn(ref, query)  # 32 x 50 x 10
 ```