rrent instance of sample being created by diffusion process.


Returns

torch.FloatTensor



the converted model output.


Convert the model output to the corresponding type that the algorithm DEIS needs.

deis_first_order_update

<
source
>
(
model_output: FloatTensor
timestep: int
prev_timestep: int
sample: FloatTensor

)
→
torch.FloatTensor

Parameters 

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


timeste