Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-cluster
Commits
fb3d8a81
Commit
fb3d8a81
authored
Feb 19, 2018
by
rusty1s
Browse files
typo
parent
fed36b1f
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
torch_cluster/functions/grid.py
torch_cluster/functions/grid.py
+1
-1
No files found.
torch_cluster/functions/grid.py
View file @
fb3d8a81
...
@@ -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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment