Add memset in else conditions, when allocating memory in spatialtf

theano/gpuarray/c_code/spatialtf_grid.c

@@ -57,8 +57,15 @@ spatialtf_grid(PyGpuArrayObject * theta,
         return -1;
     }
 
-    if ( NULL == *grid )
+    if ( NULL == *grid ||
+         PyGpuArray_NDIM( *grid ) != 4 ||
+         PyGpuArray_DIM( *grid, 0 ) != num_images ||
+         PyGpuArray_DIM( *grid, 1 ) != height ||
+         PyGpuArray_DIM( *grid, 2 ) != width  ||
+         PyGpuArray_DIM( *grid, 3) != 2 )
     {
+        Py_XDECREF( *grid );
+
         *grid = pygpu_zeros( 4, &(grid_dims[0]), theta->ga.typecode, GA_C_ORDER,
             gpu_ctx, Py_None );
         if ( NULL == *grid )
@@ -68,6 +75,10 @@ spatialtf_grid(PyGpuArrayObject * theta,
             return -1;
         }
     }
+    else
+    {
+        GpuArray_memset( &( (*grid)->ga ), 0 );
+    }
 
     const void * theta_data = PyGpuArray_DEV_DATA( theta );
     void * grid_data = PyGpuArray_DEV_DATA( *grid );

theano/gpuarray/c_code/spatialtf_sampler.c

@@ -147,6 +147,10 @@ spatialtf_sampler(PyGpuArrayObject * input,
             return -1;
         }
     }
+    else
+    {
+        GpuArray_memset( &( (*output)->ga ), 0 );
+    }
 
     const void * input_data = PyGpuArray_DEV_DATA( input );
     const void * grid_data = PyGpuArray_DEV_DATA( grid );