Skip to content

Commit b8c479d

Browse files
committed
Merge branch 'ar/add-pool-layers' of https://github.com/JuliaGraphs/GraphNeuralNetworks.jl into ar/add-pool-layers
2 parents c048532 + e80872c commit b8c479d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

GNNLux/src/layers/pool.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,4 +130,4 @@ function TopKPool(adj::AbstractMatrix, k::Int, in_channel::Int; init = glorot_un
130130
TopKPool(adj, k, init(in_channel), similar(adj, k, k))
131131
end
132132

133-
(t::TopKPool)(x::AbstractArray, ps, st) = GNNlib.topk_pool(t, x)
133+
(t::TopKPool)(x::AbstractArray, ps, st) = GNNlib.topk_pool(t, x), st

0 commit comments

Comments
 (0)