: FloatTensor
generator: typing.Optional[torch._C.Generator] = None
return_dict: bool = True

)
→
SdeVeOutput or tuple

Parameters 

model_output (torch.FloatTensor) — direct output from learned diffusion model.


sample (torch.FloatTensor) —
current instance of sample being created by diffusion process.
generator — random number generator.


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


Returns

SdeVeOutpu