parameters: (self, current_times, state)
docstring: Compute a batch of single-step predictions.
  
      Args:
        current_times: A [batch size] Tensor of times for each observation.
        state: Model state, imputed to one step before current_times.
      Returns:
        A tuple of (updated state, outputs):
          updated state: Model state updated to current_times.
          outputs: A dictionary of Tensors with keys corresponding to
              self._predict_output_names.
docstring_summary: Compute a batch of single-step predictions.
docstring_tokens: Compute a batch of single - step predictions .
function: def _prediction_step(self, current_times, state):
      """Compute a batch of single-step predictions.
  
      Args:
        current_times: A [batch size] Tensor of times for each observation.
        state: Model state, imputed to one step before current_times.
      Returns:
        A tuple of (updated state, outputs):
          updated state: Model state updated to current_times.
          outputs: A dictionary of Tensors with keys corresponding to
              self._predict_output_names.
      """
      pass
function_tokens: def _prediction_step ( self , current_times , state ) : pass