Add TODO

src/pystencils_autodiff/backends/_torch_native.py

@@ -73,6 +73,8 @@ def create_autograd_function(autodiff_obj, use_cuda):
         output_tensors = OrderedDict({f.name:
                                       field_to_tensor_dict.get(f, kwargs[f.name])
                                       for f in autodiff_obj.forward_output_fields})
+
+        # TODO: only if any of input tensors requires grad
         for o in output_tensors.values():
             if isinstance(o, torch.Tensor):
                 o.requires_grad = True