ing and saving functionality via the SchedulerMixin.save_pretrained() and
from_pretrained() functions.
For more details, see the original paper: https://arxiv.org/abs/2111.14822

log_Q_t_transitioning_to_known_class

<
source
>
(
t: torch.int32
x_t: LongTensor
log_onehot_x_t: FloatTensor
cumulative: bool

)
→
torch.FloatTensor of shape (batch size, num classes - 1, num latent pixels)

Parameters 

t (torch.Long) —
The timestep that determines w