- masked_inplace_autoregression(model, self.batch_size, result, ar_mask)
+ x, order = shuffle(result, self.height * self.width)
+ masked_inplace_autoregression(
+ model, self.batch_size, x, ar_mask, order=order
+ )
+ result = reorder(x, order, reverse=True)