def forward(self, x, emb):
    '\n        Apply the block to a Tensor, conditioned on a timestep embedding.\n\n        :param x: an [N x C x ...] Tensor of features.\n        :param emb: an [N x emb_channels] Tensor of timestep embeddings.\n        :return: an [N x C x ...] Tensor of outputs.\n        '
    return checkpoint(self._forward, (x, emb), self.parameters(), self.use_checkpoint)
Apply the block to a Tensor, conditioned on a timestep embedding.

:param x: an [N x C x ...] Tensor of features.
:param emb: an [N x emb_channels] Tensor of timestep embeddings.
:return: an [N x C x ...] Tensor of outputs.
def forward(self, x, emb):
    '\n        Apply the block to a Tensor, conditioned on a timestep embedding.\n\n        :param x: an [N x C x ...] Tensor of features.\n        :param emb: an [N x emb_channels] Tensor of timestep embeddings.\n        :return: an [N x C x ...] Tensor of outputs.\n        '
    return checkpoint(self._forward, (x, emb), self.parameters(), self.use_checkpoint)