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
af96c80a
Commit
af96c80a
authored
Mar 25, 2018
by
rusty1s
Browse files
added graclus boilerplate
parent
b8b4ae63
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
18 additions
and
1 deletion
+18
-1
test/test_graclus.py
test/test_graclus.py
+11
-0
torch_cluster/__init__.py
torch_cluster/__init__.py
+5
-1
torch_cluster/functions/graclus.py
torch_cluster/functions/graclus.py
+2
-0
No files found.
test/test_graclus.py
0 → 100644
View file @
af96c80a
import
pytest
import
torch
from
torch_cluster
import
graclus_cluster
def
test_graclus
():
edge_index
=
torch
.
LongTensor
([[
0
,
0
,
0
,
1
,
2
,
3
,
3
,
3
,
4
,
5
,
5
,
5
,
6
,
6
],
[
2
,
3
,
6
,
5
,
0
,
0
,
4
,
5
,
3
,
1
,
3
,
6
,
0
,
3
]])
edge_attr
=
torch
.
Tensor
([
2
,
2
,
2
,
1
,
2
,
2
,
2
,
2
,
2
,
1
,
2
,
2
,
2
,
2
])
graclus_cluster
(
edge_index
,
edge_attr
)
torch_cluster/__init__.py
View file @
af96c80a
from
.functions.grid
import
sparse_grid_cluster
,
dense_grid_cluster
from
.functions.graclus
import
graclus_cluster
__version__
=
'0.2.6'
__all__
=
[
'sparse_grid_cluster'
,
'dense_grid_cluster'
,
'__version__'
]
__all__
=
[
'sparse_grid_cluster'
,
'dense_grid_cluster'
,
'graclus_cluster'
,
'__version__'
]
torch_cluster/functions/graclus.py
0 → 100644
View file @
af96c80a
def
graclus_cluster
(
edge_index
,
edge_attr
=
None
,
batch
=
None
):
pass
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