fix(convert): torch.from_numpy requires positive strides

ZettaAI · Sep 6, 2023 · c616460 · c616460
1 parent 59a3109
commit c616460
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/zetta_utils/tensor_ops/convert.py b/zetta_utils/tensor_ops/convert.py
@@ -44,6 +44,8 @@ def to_torch(data: Tensor, device: torch.types.Device = None) -> torch.Tensor:
             if data.max() > np.uint64(2 ** 63 - 1):
                 raise ValueError("Unable to convert uint64 dtype to int64")
             data = data.astype(np.int64)
+        if any(v < 0 for v in data.strides):  # torch.from_numpy does not support negative strides
+            data = data.copy("K")
         result = torch.from_numpy(data).to(device)  # type: ignore # pytorch bug
 
     return result