for t in range(T-1, -1, -1):
z = torch.zeros(x.size()) if t == 0 else torch.randn(x.size())
input = torch.cat((x, torch.ones(x.size(0), 1) * 2 * t / T - 1), 1)
- x = 1 / alpha[t].sqrt() * (x - (1 - alpha[t])/(1 - alpha_bar[t]).sqrt() * model(input)) + sigma[t] * z
+ x = 1 / alpha[t].sqrt() * (x - (1 - alpha[t])/(1 - alpha_bar[t]).sqrt() * model(input)) \
+ + sigma[t] * z
######################################################################
# Plot