We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 434e1ae + ca33a1c commit 8742cf6Copy full SHA for 8742cf6
src/layers/conv.jl
@@ -454,8 +454,12 @@ Crystal Graph Convolutional network. Uses both node and edge features.
454
455
# Usage
456
457
-You can call `CGConv` in several different ways, including
458
-- With FeaturedGraph
+You can call `CGConv` in several different ways:
+
459
+- Pass a FeaturedGraph: `CGConv(fg)`, returns `FeaturedGraph`
460
+- Pass both node and edge features: `CGConv(X, E)`
461
+- Pass one matrix, which can either be node features or edge features: `CGConv(M; edge)`:
462
+ `edge` is default false, meaning that `M` denotes node features.
463
"""
464
struct CGConv{V <: AbstractFeaturedGraph, T,
465
A <: AbstractMatrix{T}, B} <: MessagePassing
0 commit comments