Update conv.jl: show

JuliaGraphs · Sep 15, 2024 · 70f16d0 · 70f16d0
1 parent bbbb07c
commit 70f16d0
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
@@ -720,4 +720,11 @@ function (l::NNConv)(g, x, e, ps, st)
     y = GNNlib.nn_conv(m, g, x, e)
     stnew = _getstate(nn)
     return y, stnew
-end
+end
+
+function Base.show(io::IO, l::NNConv)
+    print(io, "NNConv($(l.nn)")
+    l.σ == identity || print(io, ", ", l.σ)
+    l.use_bias || print(io, ", use_bias=false")
+    print(io, ")")
+end