contiguous check

sparseconvnet/inputBatch.py

@@ -25,7 +25,7 @@ class InputBatch(SparseConvNetTensor):
     def set_location(self, location, vector, overwrite=False):
         assert location.min() >= 0 and (self.spatial_size - location).min() > 0
         self.metadata.setInputSpatialLocation(
-            self.features, location, vector, overwrite)
+            self.features, location.contiguous(), vector.contiguous(), overwrite)
 
     def set_location_(self, location, vector, overwrite=False):
         self.metadata.setInputSpatialLocation(
@@ -53,7 +53,7 @@ class InputBatch(SparseConvNetTensor):
         l = locations[:, :self.dimension]
         assert l.min() >= 0 and (self.spatial_size.expand_as(l) - l).min() > 0
         self.metadata.setInputSpatialLocations(
-            self.features, locations, vectors, overwrite)
+            self.features, locations.contiguous(), vectors.contiguous(), overwrite)
 
     def set_locations_(self, locations, vector, overwrite=False):
         self.metadata.setInputSpatialLocations(