fast fixing a typo in three_nn

"known" and "unknown"
torch-points3d · May 24, 2021 · 5133e21 · 5133e21
1 parent aed9cf5
commit 5133e21
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_points_kernels/torchpoints.py b/torch_points_kernels/torchpoints.py
@@ -43,9 +43,9 @@ def three_nn(unknown, known):
     Parameters
     ----------
     unknown : torch.Tensor
-        (B, n, 3) tensor of known features
+        (B, n, 3) tensor of unknown features
     known : torch.Tensor
-        (B, m, 3) tensor of unknown features
+        (B, m, 3) tensor of known features
 
     Returns
     -------