Skip to content

Commit e012247

Browse files
committed
fix: Dimension consistency
1 parent 355d352 commit e012247

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

GNNlib/src/utils.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ function softmax_edge_neighbors(g::AbstractGNNGraph, e)
9191
end
9292
s, t = edge_index(g)
9393
if isempty(t)
94-
return zero(eltype(e))
94+
return zeros_like(e, eltype(e), (size(e)[1:end-1]..., 0))
9595
end
9696
max_ = gather(scatter(max, e, t), t)
9797
num = exp.(e .- max_)

0 commit comments

Comments
 (0)