 into images. This model inherits from ModelMixin. Check the superclass documentation for it’s generic methods implemented
for all models (such as downloading or saving).  forward  < source > ( sample: FloatTensor mask: Optional = None sample_posterior: bool = False return_dict: bool = True generator: Optional = None  )   Parameters   sample (torch.FloatTensor) — Input sample.   mask (torch.FloatTensor, optional, defaults to None) — Optional in