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

Fix dimensional error in test

This commit is contained in:
JohnnyChen 2018-09-26 22:03:38 +08:00
parent b20ae0546b
commit 3bf18347e0
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test/layers/basic.jl
View File

@ -13,9 +13,9 @@ using Test, Random
@test_throws MethodError Dense(10, 5)(1) # avoid broadcasting
@test_throws MethodError Dense(10, 5).(randn(10)) # avoid broadcasting
@test Dense(10, 1, identity, initW = ones, initb = zeros)(ones(10,1)) == [10]
@test Dense(10, 1, identity, initW = ones, initb = zeros)(ones(10,2)) == [10 10]
@test Dense(10, 2, identity, initW = ones, initb = zeros)(ones(10,1)) == [10; 10]
@test Dense(10, 1, identity, initW = ones, initb = zeros)(ones(10,1)) == 10*ones(1, 1)
@test Dense(10, 1, identity, initW = ones, initb = zeros)(ones(10,2)) == 10*ones(1, 2)
@test Dense(10, 2, identity, initW = ones, initb = zeros)(ones(10,1)) == 10*ones(2, 1)
@test Dense(10, 2, identity, initW = ones, initb = zeros)([ones(10,1) 2*ones(10,1)]) == [10 20; 10 20]
end