Skip to content

Commit a625ce1

Browse files
authored
Fix: fixed GMMconv implementation (#645)
Signed-off-by: Parvm1102 <parvmittal31757@gmail.com>
1 parent c4c69f4 commit a625ce1

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

GNNlib/src/layers/conv.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -378,7 +378,7 @@ function gmm_conv(l, g::GNNGraph, x::AbstractMatrix, e::AbstractMatrix)
378378
w = reshape(e, (ein, 1, num_edges))
379379
mu = reshape(l.mu, (ein, l.K, 1))
380380

381-
w = @. ((w - mu)^2) / 2
381+
w = @. -((w - mu)^2) / 2
382382
w = w .* reshape(l.sigma_inv .^ 2, (ein, l.K, 1))
383383
w = exp.(sum(w, dims = 1)) # (1, K, num_edge)
384384

0 commit comments

Comments
 (0)