estep: torch.int64 sample: LongTensor generator: Optional = None return_dict: bool = True  ) → VQDiffusionSchedulerOutput or tuple  Parameters   t (torch.long) —
The timestep that determines which transition matrices are used.   x_t (torch.LongTensor of shape (batch size, num latent pixels)) —
The classes of each latent pixel at time t.   generator (torch.Generator, or None) —
A random number generator for the noise applied to p(x_{t-1} | x_t) 