Merge pull request #6376 from nouiz/broadcast_corrmm

theano/gpuarray/blas.py

@@ -1033,7 +1033,7 @@ class GpuCorrMM_gradWeights(BaseGpuCorrMM):
             assert shape[1].ndim == 0
 
         if self.unshared:
-            broadcastable = [topgrad.type.broadcastable[0], False, False,
+            broadcastable = [topgrad.type.broadcastable[1], False, False,
                              img.type.broadcastable[1], False, False]
         else:
             broadcastable = [topgrad.type.broadcastable[1], img.type.broadcastable[1],

theano/tensor/nnet/corr.py

@@ -695,10 +695,10 @@ class CorrMM_gradWeights(BaseCorrMM):
             height_width = [as_tensor_variable(shape[0]).astype('int64'), as_tensor_variable(shape[1]).astype('int64')]
 
         if self.unshared is True:
-            broadcastable = [topgrad.type.broadcastable[0], False, False,
+            broadcastable = [topgrad.type.broadcastable[1], False, False,
                              img.type.broadcastable[1], False, False]
         else:
-            broadcastable = [topgrad.type.broadcastable[0], img.type.broadcastable[1],
+            broadcastable = [topgrad.type.broadcastable[1], img.type.broadcastable[1],
                              False, False]
         dtype = img.type.dtype
         return Apply(self, [img, topgrad] + height_width,