ion chain.   sample (torch.FloatTensor) —
A current instance of a sample created by the diffusion process.   return_dict (bool) —
Whether or not to return a SchedulerOutput or tuple.   Returns 
SchedulerOutput or tuple
  
If return_dict is True, SchedulerOutput is returned, otherwise a
tuple is returned where the first element is the sample tensor.
  Predict the sample from the previous timestep by reversing the SDE. This function propagates th