save_image(x, "output.png")
# Generate a bunch of images
z = sample_gaussian(mean_p_Z.expand(x.size(0), -1), log_var_p_Z.expand(x.size(0), -1))
mean_p_X_given_z, log_var_p_X_given_z = model_p_X_given_z(z)
save_image(x, "output.png")
# Generate a bunch of images
z = sample_gaussian(mean_p_Z.expand(x.size(0), -1), log_var_p_Z.expand(x.size(0), -1))
mean_p_X_given_z, log_var_p_X_given_z = model_p_X_given_z(z)