change infer_shape in corr3dmm for grouped convolutions

theano/tensor/nnet/corr3d.py

@@ -662,6 +662,7 @@ class Corr3dMM_gradWeights(BaseCorr3dMM):
         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_depth = node.inputs[-3:]
         if ((dH != 1) or (padH == -1)):
@@ -773,6 +774,7 @@ class Corr3dMM_gradInputs(BaseCorr3dMM):
         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_depth = node.inputs[-3:]
         if padH == -1: