TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test repeat fix

Browse Source
This commit is contained in:
Mike Innes 2018-08-23 11:29:43 +01:00
parent 6c355e93d2
commit dfe7578216
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
test/tracker.jl
View File

@ -65,7 +65,7 @@ end
@test gradtest(vcatf, rand(5)', rand(2,5)) @test gradtest(vcatf, rand(5)', rand(2,5))
end end
@testset for hcatf in [hcat, cat2] @testset for hcatf in [hcat, cat2]
@test gradtest(hcatf, rand(5), rand(5)) @test gradtest(hcatf, rand(5), rand(5))
@test gradtest(hcatf, rand(5)', rand(5)') @test gradtest(hcatf, rand(5)', rand(5)')
@ -117,6 +117,7 @@ end
@test gradtest(x -> permutedims(x, [3,1,2]), rand(4,5,6)) @test gradtest(x -> permutedims(x, [3,1,2]), rand(4,5,6))
@test gradtest(x -> repeat(x; inner=2), rand(5))
@test gradtest(x -> repeat(x; inner=2, outer=3), rand(5)) @test gradtest(x -> repeat(x; inner=2, outer=3), rand(5))
@test gradtest(x -> repeat(x; inner=(2,2,1), outer=(1,1,3)), rand(5,4,3)) @test gradtest(x -> repeat(x; inner=(2,2,1), outer=(1,1,3)), rand(5,4,3))