Skip to content

Commit

Permalink
vcat test FluxML#213
Browse files Browse the repository at this point in the history
Co-authored-by: improbable22 <improbable+github@gmail.com>

# Conflicts:
#	test/tracker.jl
  • Loading branch information
gustafsson committed May 2, 2018
1 parent 1c189c6 commit ca12dc1
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions test/tracker.jl
Original file line number Diff line number Diff line change
Expand Up @@ -72,6 +72,11 @@ end
end

@test gradtest((a,b)->cat((2,3,5), a, b), rand(2,3), rand(2,4,2,1))

@testset "issue #213" begin
A, B, C = rand(2,2), rand(2,2), rand(2,2)
@test vcat(A, B, C |> param) == vcat(param.((A,B,C))...)
end
end

@test gradtest(x -> permutedims(x, [3,1,2]), rand(4,5,6))
Expand Down

0 comments on commit ca12dc1

Please sign in to comment.