original_docstring: r"""Enforce the output of a neural network to satisfy the initial condition.
  
          :param net: The neural network that approximates the ODE.
          :type net: `torch.nn.Module`
          :param t: The points where the neural network output is evaluated.
          :type t: `torch.tensor`
          :return: The modified output which now satisfies the initial condition.
          :rtype: `torch.tensor`
  
          .. note::
              `enforce` is meant to be called by the function `solve` and `solve_system`.
          
docstring: r"""Enforce the output of a neural network to satisfy the initial condition.
docstring_tokens: r " " " Enforce the output of a neural network to satisfy the initial condition .
code: def enforce(self, net, t):
          r"""Enforce the output of a neural network to satisfy the initial condition.
  
          :param net: The neural network that approximates the ODE.
          :type net: `torch.nn.Module`
          :param t: The points where the neural network output is evaluated.
          :type t: `torch.tensor`
          :return: The modified output which now satisfies the initial condition.
          :rtype: `torch.tensor`
  
          .. note::
              `enforce` is meant to be called by the function `solve` and `solve_system`.
          """
          x = _network_output(net, t, self.ith_unit)
          if self.x_0_prime:
              return self.x_0 + (t-self.t_0)*self.x_0_prime + ( (1-torch.exp(-t+self.t_0))**2 )*x
          else:
              return self.x_0 + (1-torch.exp(-t+self.t_0))*x
code_tokens: def enforce ( self , net , t ) : x = _network_output ( net , t , self . ith_unit ) if self . x_0_prime : return self . x_0 + ( t - self . t_0 ) * self . x_0_prime + ( ( 1 - torch . exp ( - t + self . t_0 ) ) ** 2 ) * x else : return self . x_0 + ( 1 - torch . exp ( - t + self . t_0 ) ) * x
short_docstring: r"""Enforce the output of a neural network to satisfy the initial condition.
short_docstring_tokens: r " " " Enforce the output of a neural network to satisfy the initial condition .
comment: r"""Enforce the output of a neural network to satisfy the initial condition.
  
          :param net: The neural network that approximates the ODE.
          :type net: `torch.nn.Module`
          :param t: The points where the neural network output is evaluated.
          :type t: `torch.tensor`
          :return: The modified output which now satisfies the initial condition.
          :rtype: `torch.tensor`
  
          .. note::
              `enforce` is meant to be called by the function `solve` and `solve_system`.
          """