
model_output: FloatTensor
timestep: int
prev_timestep: int
sample: FloatTensor

)
→
torch.FloatTensor

Parameters 

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


timestep (int) — current discrete timestep in the diffusion chain.


prev_timestep (int) — previous discrete timestep in the diffusion chain.


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


Returns

tor