fix error of gradient over scalar matrix multiplication

FluxML · Dec 12, 2021 · 21e7f7f · 21e7f7f
1 parent 3b838d8
commit 21e7f7f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/layers/conv.jl b/src/layers/conv.jl
@@ -104,7 +104,7 @@ function (c::ChebConv)(fg::FeaturedGraph, X::AbstractMatrix{T}) where T
     Y = view(c.weight,:,:,1) * Z_prev
     Y += view(c.weight,:,:,2) * Z
     for k = 3:c.k
-        Z, Z_prev = 2*Z*L̃ - Z_prev, Z
+        Z, Z_prev = 2 .* Z * L̃ - Z_prev, Z
         Y += view(c.weight,:,:,k) * Z
     end
     return Y .+ c.bias