ters   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, the single step transition matrix t-1->t is used. If cumulative is
True, the cu