fix: workaround negative strides issue when converting to torch tensor

krishnanlab · Oct 17, 2023 · a2a944e · a2a944e
1 parent 38899d5
commit a2a944e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/obnb/transform/node_feature.py b/src/obnb/transform/node_feature.py
@@ -127,7 +127,9 @@ class SVD(BaseNodeFeatureTransform):
     def _prepare_feat(self, dataset):
         sparse_adj = sp.csr_matrix(dataset.get_adj())
         feat, _, _ = sp.linalg.svds(sparse_adj, k=self.dim, which="LM")
-        return feat
+        # Work around for potential negative strides by making a copy of the array
+        # https://discuss.pytorch.org/t/negative-strides-in-tensor-error/134287/2
+        return feat.copy()
 
 
 @register_nodefeat