Assert input.ndim == 5 in pool3d2d reference.

theano/gpuarray/tests/test_dnn.py

@@ -172,6 +172,7 @@ def pool3d2d(input, ds=(2, 2, 2), strides=None, pad=(0, 0, 0),
     if strides is None:
         strides = ds
 
+    assert input.ndim == 5
     shape = input.shape
 
     # reshape to B, C*0, 1, 2 and do the pooling on 1, 2

theano/sandbox/cuda/tests/test_dnn.py

@@ -174,6 +174,7 @@ def pool3d2d(input, ds=(2, 2, 2), strides=None, pad=(0, 0, 0),
     if strides is None:
         strides = ds
 
+    assert input.ndim == 5
     shape = input.shape
 
     # resahpe to B, C*0, 1, 2 and do the pooling on 1, 2