Change vmap randomness to 'same' as 'different' seems to be not suppo…

…rted.
Tim-Salzmann · May 29, 2024 · 4394d56 · 4394d56
1 parent c135d4f
commit 4394d56
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/l4casadi/realtime/sensitivities.py b/l4casadi/realtime/sensitivities.py
@@ -32,7 +32,9 @@ def batched_jacobian(func: Callable, inputs: torch.Tensor, create_graph=False, r
     if inputs.shape[0] == 1:
         vmap_randomness = 'same'
     else:
-        vmap_randomness = 'different'
+        # https://github.com/pytorch/functorch/issues/996
+        # Should be 'different'
+        vmap_randomness = 'same'
 
     if not create_graph:
         with torch.no_grad():
@@ -65,7 +67,9 @@ def batched_hessian(func: Callable, inputs: torch.Tensor, create_graph=False,
     if inputs.shape[0] == 1:
         vmap_randomness = 'same'
     else:
-        vmap_randomness = 'different'
+        # https://github.com/pytorch/functorch/issues/996
+        # Should be 'different'
+        vmap_randomness = 'same'
 
     def aux_function_jac(func):
         def inner_aux(inputs):