Commit fb3d8a81 authored by rusty1s's avatar rusty1s
Browse files

typo

parent fed36b1f
...@@ -21,7 +21,7 @@ def grid_cluster(position, size, batch=None, offset=None, fake_nodes=False): ...@@ -21,7 +21,7 @@ def grid_cluster(position, size, batch=None, offset=None, fake_nodes=False):
position = torch.cat([batch, position], dim=-1) position = torch.cat([batch, position], dim=-1)
size = torch.cat([size.new(1).fill_(1), size], dim=-1) size = torch.cat([size.new(1).fill_(1), size], dim=-1)
# Translate to minimal positive positions. # Translate to minimal positive positions if no offset is passed.
if offset is None: if offset is None:
min = position.min(dim=-2, keepdim=True)[0] min = position.min(dim=-2, keepdim=True)[0]
position = position - min position = position - min
......
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