testing infer_shape: Op Rebroadcast

theano/tensor/tests/test_basic.py

@@ -35,7 +35,7 @@ from theano.tensor import (_shared, wvector, bvector, autocast_float_as,
         iscalars, arange,  dscalars, fvector, imatrix, numeric_grad,
         opt, ComplexError, TensorDot, lvector, true_div, max, min, Split, roll,
         tile, patternbroadcast, Eye, Shape, Default, Dot, PermuteRowElements,
-        ScalarFromTensor, TensorFromScalar)
+        ScalarFromTensor, TensorFromScalar, dtensor4, Rebroadcast)
 from theano.tests import unittest_tools as utt
 from theano.printing import debugprint
 
@@ -6155,6 +6155,14 @@ class TestInferShape(utt.InferShapeTester):
                                 [TensorFromScalar()(aiscal)],
                         [4.], TensorFromScalar)
 
+        # Rebroadcast:
+        adtens4 = dtensor4()
+        adict = [(0, False), (1, True), (2, False), (3, True)]
+        adtens4_val = rand(2, 1, 3, 1)
+        self._compile_and_check([adtens4],
+                                [Rebroadcast(*adict)(adtens4)],
+                                [adtens4_val], Rebroadcast)
+        
 if __name__ == '__main__':
 
     t = TestInferShape('setUp')