Skip to content

Commit f1cd90c

Browse files
Update src/layers/conv.jl
Co-authored-by: Carlo Lucibello <[email protected]>
1 parent 2e6a392 commit f1cd90c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/layers/conv.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1098,13 +1098,13 @@ t = [2,3,1,1]
10981098
g = GNNGraph(s,t)
10991099
in_feature, out_feature, n_k, e_dim = 4, 7, 8, 10
11001100
x = randn(in_feature, g.num_nodes)
1101-
u = randn(e_dim, g.num_edges)
1101+
e = randn(e_dim, g.num_edges)
11021102
11031103
# create layer
11041104
l = GMMConv(in_feature=>out_feature, n_k, e_dim)
11051105
11061106
# forward pass
1107-
l(g, x, u)
1107+
l(g, x, e)
11081108
```
11091109
"""
11101110

0 commit comments

Comments
 (0)