fix
This commit is contained in:
parent
27a9a7b9cf
commit
73c5d9f25c
|
@ -63,21 +63,21 @@ end
|
|||
end
|
||||
|
||||
@testset "Conv with non quadratic window #700" begin
|
||||
data = fill(0f0, 7,7,1,1)
|
||||
data = zeros(Float32, 7,7,1,1)
|
||||
data[4,4,1,1] = 1
|
||||
|
||||
l = Conv((3,3), 1=>1)
|
||||
expected = fill(zero(eltype(l.weight)),5,5,1,1)
|
||||
expected = zeros(eltype(l.weight),5,5,1,1)
|
||||
expected[2:end-1,2:end-1,1,1] = l.weight
|
||||
@test expected == l(data)
|
||||
|
||||
l = Conv((3,1), 1=>1)
|
||||
expected = fill(zero(eltype(l.weight)),5,7,1,1)
|
||||
expected = zeros(eltype(l.weight),5,7,1,1)
|
||||
expected[2:end-1,4,1,1] = l.weight
|
||||
@test_broken expected == l(data)
|
||||
|
||||
l = Conv((1,3), 1=>1)
|
||||
expected = fill(zero(eltype(l.weight)),7,5,1,1)
|
||||
expected = zeros(eltype(l.weight),7,5,1,1)
|
||||
expected[4,2:end-1,1,1] = l.weight
|
||||
@test_broken expected == l(data)
|
||||
|
||||
|
|
Loading…
Reference in New Issue