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

update test

Browse Source
This commit is contained in:
chengchingwen 2019-03-14 21:55:37 +08:00
parent 934f0840b2
commit 59da68b4d9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/layers/normalisation.jl
View File

@ -29,10 +29,10 @@ using Flux.Tracker: data
x = rand(100, 50) x = rand(100, 50)
m = Dropout(0.5) m = Dropout(0.5)
y = m(x, (100, 1)) y = m(x, 2)
c = map(i->count(a->a==0, @view y[:, i]), 1:50) c = map(i->count(a->a==0, @view y[:, i]), 1:50)
@test minimum(c) == maximum(c) @test minimum(c) == maximum(c)
y = m(x, (1, 50)) y = m(x, 1)
c = map(i->count(a->a==0, @view y[i, :]), 1:100) c = map(i->count(a->a==0, @view y[i, :]), 1:100)
@test minimum(c) == maximum(c) @test minimum(c) == maximum(c)
end end