The direct output from learned diffusion model.   sigma_hat (float) — TODO   sigma_prev (float) — TODO   sample_hat (torch.FloatTensor) — TODO   sample_prev (torch.FloatTensor) — TODO   derivative (torch.FloatTensor) — TODO   return_dict (bool, optional, defaults to True) —