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

another small test

Browse Source
This commit is contained in:
ayush-1506 2019-05-09 16:43:28 +05:30
parent 9a3aa18c17
commit 99d07e67db
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test/layers/conv.jl
View File

@ -66,9 +66,8 @@ end
x = rand(Float32, 28, 28, 1, 1) x = rand(Float32, 28, 28, 1, 1)
w = rand(2,2,1,1) w = rand(2,2,1,1)
y = CrossCor(w, [0.0]) y = CrossCor(w, [0.0])
x_pred = y(x)
@test sum(w .* x[1:2, 1:2, :, :]) == x_pred[1, 1, 1, 1] @test sum(w .* x[1:2, 1:2, :, :]) == y(x)[1, 1, 1, 1]
r = zeros(Float32, 28, 28, 1, 5) r = zeros(Float32, 28, 28, 1, 5)
m = Chain( m = Chain(
@ -80,6 +79,8 @@ end
Dense(288, 10), softmax) Dense(288, 10), softmax)
@test size(m(r)) == (10, 5) @test size(m(r)) == (10, 5)
@test y(x) != Conv(w, [0.0])(x)
@test CrossCor(w[end:-1:1, end:-1:1, :, :], [0.0])(x) == Conv(w, [0.0])(x)
end end
@testset "Conv with non quadratic window #700" begin @testset "Conv with non quadratic window #700" begin