orch.FloatTensor) —
current instance of sample being created by diffusion process.


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


Returns

~scheduling_utils.SchedulerOutput or tuple



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


Step function propagating the sample with the linear multi-step method. This has 