[core] Fix exception when passing FlowField to F.grid_sample()

BioMedIA · Dec 14, 2023 · b6a3097 · b6a3097
1 parent a4b1359
commit b6a3097
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/deepali/data/flow.py b/src/deepali/data/flow.py
@@ -142,8 +142,6 @@ def _torch_function_result(
 
     @classmethod
     def __torch_function__(cls, func, types, args=(), kwargs=None):
-        if func == F.grid_sample:
-            raise ValueError("Argument of F.grid_sample() must be a batch, not a single image")
         if kwargs is None:
             kwargs = {}
         data = Tensor.__torch_function__(func, (Tensor,), args, kwargs)