batch_size, sequence_length, feature_dimension). Apply the Transformer blocks in the standard way. Reshape to image. When the input is discrete: It is assumed one of the input classes is the masked latent pixel. The predicted classes of the unnoised image don’t contain a prediction for the masked pixel because the unnoised image cannot be masked. Convert input (classes of latent pixels) to embeddings and apply positional embeddings. Apply the T