ndom number generator.


return_dict (bool) — option for returning tuple rather than SchedulerOutput class


Returns

SdeVeOutput or tuple



SdeVeOutput if
return_dict is True, otherwise a tuple. When returning a tuple, the first element is the sample tensor.


Correct the predicted sample based on the output model_output of the network. This is often run repeatedly
after making the prediction for the previous timestep.

step_pred

<
source
>
