Merge pull request #522 from kolia/tiny_stack_bugfix

Tiny bugfix: `stack` was still calling julia 0.6 `cat`
This commit is contained in:
Dhairya Gandhi 2019-02-15 20:50:08 +05:30 committed by GitHub
commit 78876a14b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -87,6 +87,12 @@ end
@test size.(params(m)) == [(5, 10), (5, 5), (5,), (5,)]
end
@testset "Basic Stacking" begin
x = randn(3,3)
stacked = stack([x, x], 2)
@test size(stacked) == (3,2,3)
end
@testset "Precision" begin
m = Chain(Dense(10, 5, relu), Dense(5, 2))
x = rand(10)