diff --git a/torchinterp1d/interp1d.py b/torchinterp1d/interp1d.py index 059338d..636bc3d 100644 --- a/torchinterp1d/interp1d.py +++ b/torchinterp1d/interp1d.py @@ -147,7 +147,7 @@ def sel(name): ynew = ynew.view(original_xnew_shape) ctx.save_for_backward(ynew, *saved_inputs) - return ynew + return ynew.detach() @staticmethod def backward(ctx, grad_out): @@ -165,4 +165,4 @@ def backward(ctx, grad_out): return (*result,) -interp1d = Interp1d.apply \ No newline at end of file +interp1d = Interp1d.apply