Skip to content

GitLab

Commit 4fbb55f7 authored by Duc Nguyen's avatar Duc Nguyen
Browse files

minors

parent 82968b99
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
torch_cluster/fps.py
View file @ 4fbb55f7
...@@ -35,8 +35,6 @@ def fps(src: torch.Tensor, batch: Optional[torch.Tensor] = None, ...@@ -35,8 +35,6 @@ def fps(src: torch.Tensor, batch: Optional[torch.Tensor] = None,
assert len(ratio.shape) < 2, 'Invalid ratio' assert len(ratio.shape) < 2, 'Invalid ratio'
ratio = ratio.to(src.device) ratio = ratio.to(src.device)
if len(ratio.shape) == 1:
assert ratio.shape[0] == int(batch.max()) + 1, 'Mismatched input and ratio numbers'
if batch is not None: if batch is not None:
assert src.size(0) == batch.numel() assert src.size(0) == batch.numel()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment