Fix PyTorch scalar reshape

tum-pbs · Dec 29, 2024 · 167980e · 167980e
1 parent 37db76d
commit 167980e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/phiml/backend/torch/_torch_backend.py b/phiml/backend/torch/_torch_backend.py
@@ -355,6 +355,8 @@ def grid_sample(self, grid, coordinates, extrapolation: str):
         return result
 
     def reshape(self, value, shape):
+        if not shape:
+            return value
         value = self.as_tensor(value)
         if value.is_contiguous():
             return value.view(*shape)