We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e40d0bd commit 836b1baCopy full SHA for 836b1ba
test/runtests.jl
@@ -155,5 +155,6 @@ end
155
grad_slicemap = Zygote.gradient(() -> sum(slicemap(F(w), x, dims=1)), Zygote.Params([w,x]))
156
@test grad_slicemap[x] ≈ gradx
157
@test_broken grad_slicemap[w] ≈ gradw # wrong numbers
158
+ @test gradw ≈ Zygote.gradient(w -> sum(slicemap(F(w), x, dims=1)), w)[1]
159
160
end
0 commit comments