LRN layer: 4D tensor -> ND-tensor

pluskid · Dec 20, 2014 · 597ab49 · 597ab49
1 parent deb64a5
commit 597ab49
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/layers/lrn.jl b/test/layers/lrn.jl
@@ -3,8 +3,7 @@ function test_lrn_layer(backend::Backend, mode::LRNModeType, tensor_dim, T, eps)
 
 
   dims = tuple((abs(rand(Int,tensor_dim)) % 6 + 6)...)
-  #op_dim = max(abs(rand(Int)) % tensor_dim, 1)
-  op_dim = 3
+  op_dim = max(abs(rand(Int)) % tensor_dim, 1)
 
   println("    > Setup with dims $dims")