   log_p_x_0 (torch.FloatTensor of shape (batch size, num classes - 1, num latent pixels)) —
The log probabilities for the predicted classes of the initial latent pixels. Does not include a
prediction for the masked class as the initial unnoised image cannot be masked.   x_t (torch.LongTensor of shape (batch size, num latent pixels)) —
The classes of each latent pixel at time t.   t (torch.Long) —
The timestep that determines which transition m