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

Merge pull request #522 from kolia/tiny_stack_bugfix

Browse Source 
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
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test/utils.jl
View File

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