atTensor of shape (batch size, num classes - 1, num latent pixels)

Parameters 

t (torch.Long) —
The timestep that determines which transition matrix is used.


x_t (torch.LongTensor of shape (batch size, num latent pixels)) —
The classes of each latent pixel at time t.


log_onehot_x_t (torch.FloatTensor of shape (batch size, num classes, num latent pixels)) —
The log one-hot vectors of x_t


cumulative (bool) —
If cumulative is False, we use