Skip to content

Commit cd59711

Browse files
committed
these are fixed
1 parent 218ad41 commit cd59711

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/gradcheck.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1157,10 +1157,10 @@ end
11571157
end
11581158

11591159
@testset "hvcat" begin
1160-
@test gradient(xs -> hvcat((2,2),xs...)[1,1], [1,2,3,4])[1] == (1,0,0,0)
1161-
@test gradient(xs -> hvcat((2,2),xs...)[2,1], [1,2,3,4])[1] == (0,0,1,0)
1162-
@test gradient(xs -> hvcat((2,2),xs...)[1,2], [1,2,3,4])[1] == (0,1,0,0)
1163-
@test gradient(xs -> hvcat((2,2),xs...)[2,2], [1,2,3,4])[1] == (0,0,0,1)
1160+
@test gradient(xs -> hvcat((2,2),xs...)[1,1], [1,2,3,4])[1] == [1,0,0,0]
1161+
@test gradient(xs -> hvcat((2,2),xs...)[2,1], [1,2,3,4])[1] == [0,0,1,0]
1162+
@test gradient(xs -> hvcat((2,2),xs...)[1,2], [1,2,3,4])[1] == [0,1,0,0]
1163+
@test gradient(xs -> hvcat((2,2),xs...)[2,2], [1,2,3,4])[1] == [0,0,0,1]
11641164
# https://github.com/FluxML/Zygote.jl/issues/513
11651165
@test gradient(x -> hvcat((2,2),1,2,3,x)[4], 4.0) == (1.0,)
11661166
end

0 commit comments

Comments
 (0)