modify infer_shape in corrmm

theano/tensor/nnet/corr.py

@@ -607,6 +607,7 @@ class CorrMM_gradWeights(BaseCorrMM):
         imshp = input_shape[0]
         topshp = input_shape[1]
         ssize, imshp = imshp[1], list(imshp[2:])
+        ssize = ssize // self.num_groups
         nkern, topshp = topshp[1], list(topshp[2:])
         height_width = node.inputs[-2:]
         if ((dH != 1) or (padH == -1)):
@@ -707,6 +708,7 @@ class CorrMM_gradInputs(BaseCorrMM):
         kshp = input_shape[0]
         topshp = input_shape[1]
         ssize, kshp = kshp[1], list(kshp[2:])
+        ssize = ssize * self.num_groups
         bsize, topshp = topshp[0], list(topshp[2:])
         height_width = node.inputs[-2:]
         if padH == -1: