on master help sign in
Add another test
eb179d0Huw Campbell
test/Test/Grenade/Layers/Pooling.hs+11
@@ -21,6 +21,17 @@ prop_pool = once $
21 out = poolForward 2 2 1 1 2 3 input
21 out = poolForward 2 2 1 1 2 3 input
22 in expected === out
22 in expected === out
23
23
24+prop_pool_rectangular = once $
25+ let input = (3><4)
26+ [ 1.0, 14.0, 3.0, 4.0
27+ , 5.0, 6.0, 7.0, 8.0
28+ , 9.0, 10.0, 11.0, 12.0 ]
29+ expected = (2><2)
30+ [ 14.0, 14.0
31+ , 11.0, 12.0 ]
32+ out = poolForward 2 3 1 1 2 2 input
33+ in expected === out
34+
24prop_pool_backwards = once $
35prop_pool_backwards = once $
25 let input = (3><4)
36 let input = (3><4)
26 [ 1.0, 2.0, 3.0, 4.0
37 [ 1.0, 2.0, 3.0, 4.0