change initial value reparameterization to previous version
@@ -28,7 +28,9 @@ class InitialValueProblem(Condition):
if self.x_0_prime:
return self.x_0 + (1-torch.exp(-t+self.t_0))*self.x_0_prime + ( (1-torch.exp(-t+self.t_0))**2 )*x
else:
- return torch.exp(-t+self.t_0)*self.x_0 + (1-torch.exp(-t+self.t_0))*x
+ # TODO: what the right re-parameterization?
+ # return torch.exp(-t+self.t_0)*self.x_0 + (1-torch.exp(-t+self.t_0))*x
+ return self.x_0 + (1-torch.exp(-t+self.t_0))*x
class TwoPointDirichletBoundaryCondition(Condition):
"""
