deep-graph-matching-consensus
latest
Package Reference
dgmc.models
dgmc.utils
deep-graph-matching-consensus
Docs
»
dgmc.models
Edit on GitHub
dgmc.models
¶
class
dgmc.models.
GIN
(
in_channels
,
out_channels
,
num_layers
,
batch_norm
=
False
,
cat
=
True
,
lin
=
True
)
[source]
¶
forward
(
x
,
edge_index
,
*
args
)
[source]
¶
reset_parameters
(
)
[source]
¶
class
dgmc.models.
SplineCNN
(
in_channels
,
out_channels
,
dim
,
num_layers
,
cat
=
True
,
lin
=
True
,
dropout
=
0.0
)
[source]
¶
forward
(
x
,
edge_index
,
edge_attr
,
*
args
)
[source]
¶
reset_parameters
(
)
[source]
¶
class
dgmc.models.
RelCNN
(
in_channels
,
out_channels
,
num_layers
,
batch_norm
=
False
,
cat
=
True
,
lin
=
True
,
dropout
=
0.0
)
[source]
¶
forward
(
x
,
edge_index
,
*
args
)
[source]
¶
reset_parameters
(
)
[source]
¶
Read the Docs
v: latest
Versions
latest
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.